Fix config test and add changelog

Noble-Lab · Nov 28, 2023 · 5e58b65 · 5e58b65
1 parent 625f7f3
commit 5e58b65
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -12,6 +12,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 - `accelerator` parameter controls the accelerator (CPU, GPU, etc) that is used.
 - `devices` parameter controls the number of accelerators used.
 - `val_check_interval` parameter controls the frequency of both validation epochs and model checkpointing during training.
+- `train_label_smoothing` parameter controls the amount of label smoothing applied when calculating the training loss.
 
 ### Changed
 

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -190,6 +190,7 @@ def tiny_config(tmp_path):
         "n_head": 2,
         "dim_feedforward": 10,
         "n_layers": 1,
+        "train_label_smoothing": 0.01,
         "warmup_iters": 1,
         "max_iters": 1,
         "max_epochs": 20,