Skip to content

Commit 165fc43

Browse files
committed
fix comment
1 parent 42cbf75 commit 165fc43

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

library/train_util.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1885,7 +1885,7 @@ def add_optimizer_arguments(parser: argparse.ArgumentParser):
18851885
"--optimizer_type",
18861886
type=str,
18871887
default="",
1888-
help="Optimizer to use / オプティマイザの種類: AdamW (default), AdamW8bit, Lion, SGDNesterov, SGDNesterov8bit, DAdaptation(DAdaptAdam), DAdaptAdaGrad, DAdaptAdan, DAdaptSGD, AdaFactor",
1888+
help="Optimizer to use / オプティマイザの種類: AdamW (default), AdamW8bit, Lion8bit, Lion, SGDNesterov, SGDNesterov8bit, DAdaptation(DAdaptAdam), DAdaptAdaGrad, DAdaptAdan, DAdaptSGD, AdaFactor",
18891889
)
18901890

18911891
# backward compatibility
@@ -2467,7 +2467,7 @@ def task():
24672467

24682468

24692469
def get_optimizer(args, trainable_params):
2470-
# "Optimizer to use: AdamW, AdamW8bit, Lion, SGDNesterov, SGDNesterov8bit, DAdaptation, DAdaptation(DAdaptAdam), DAdaptAdaGrad, DAdaptAdan, DAdaptSGD, Adafactor"
2470+
# "Optimizer to use: AdamW, AdamW8bit, Lion, SGDNesterov, SGDNesterov8bit, Lion8bit, DAdaptation, DAdaptation(DAdaptAdam), DAdaptAdaGrad, DAdaptAdan, DAdaptSGD, Adafactor"
24712471

24722472
optimizer_type = args.optimizer_type
24732473
if args.use_8bit_adam:

0 commit comments

Comments
 (0)