diff options
Diffstat (limited to 'train_ti.py')
-rw-r--r-- | train_ti.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/train_ti.py b/train_ti.py index ee65b44..036c288 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -293,8 +293,8 @@ def parse_args(): | |||
293 | parser.add_argument( | 293 | parser.add_argument( |
294 | "--optimizer", | 294 | "--optimizer", |
295 | type=str, | 295 | type=str, |
296 | default="adam", | 296 | default="dadan", |
297 | help='Optimizer to use ["adam", "adam8bit"]' | 297 | help='Optimizer to use ["adam", "adam8bit", "dadam", "dadan"]' |
298 | ) | 298 | ) |
299 | parser.add_argument( | 299 | parser.add_argument( |
300 | "--adam_beta1", | 300 | "--adam_beta1", |