NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-22 16:48:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

cleanup 3

Browse source
This commit is contained in:
Jai Suphavadeeprasit 2026-02-13 12:39:37 -05:00
parent 39d307b440
commit 43cc71e070
4 changed files with 4 additions and 93 deletions
Download patch file Download diff file Expand all files Collapse all files

4
example_trainer/cli.py
View file

@ -65,10 +65,10 @@ def add_training_args(parser: argparse.ArgumentParser) -> None:
group.add_argument(
"--optimizer",
type=str,
choices=["adamw", "adamw_8bit", "adamw_cpu", "adafactor"],
choices=["adamw", "adamw_8bit", "adafactor"],
default="adamw_8bit",
help="Optimizer: 'adamw' (full precision), 'adamw_8bit' (8-bit states), "
"'adamw_cpu' (CPU offload), 'adafactor' (no momentum)",
"'adafactor' (no momentum)",
)
group.add_argument(
"--device",