diff options
| author | Kawrakow <48489457+ikawrakow@users.noreply.github.com> | 2023-07-25 18:35:53 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-25 18:35:53 +0300 |
| commit | eb542d39324574a6778fad9ba9e34ba7a14a82a3 (patch) | |
| tree | 3009fde3ceb24f19dfbb2da7c072942ba06eaa83 /examples/train-text-from-scratch | |
| parent | 07aaa0f63fccaeab099b3a732abda20b921bc5a5 (diff) | |
Add LLAMA_DEFAULT_RMS_EPS so we can change the default (#2384)
Co-authored-by: Iwan Kawrakow <iwan.kawrakow@gmail.com>
Diffstat (limited to 'examples/train-text-from-scratch')
| -rw-r--r-- | examples/train-text-from-scratch/train-text-from-scratch.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/train-text-from-scratch/train-text-from-scratch.cpp b/examples/train-text-from-scratch/train-text-from-scratch.cpp index 4bbf6b7..54dc2be 100644 --- a/examples/train-text-from-scratch/train-text-from-scratch.cpp +++ b/examples/train-text-from-scratch/train-text-from-scratch.cpp @@ -16,7 +16,7 @@ #pragma warning(disable: 4244 4267) // possible loss of data #endif -static const float rms_norm_eps = 1e-6f; +static const float rms_norm_eps = LLAMA_DEFAULT_RMS_EPS; struct random_normal_distribution { std::mt19937 gen; |