mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-08-29 02:21:18 +08:00
udpate
This commit is contained in:
parent
0df9491ff0
commit
cae3705109
@ -104,7 +104,7 @@ def build_parser():
|
|||||||
"--optimizer",
|
"--optimizer",
|
||||||
type=str,
|
type=str,
|
||||||
choices=["adam", "adamw"],
|
choices=["adam", "adamw"],
|
||||||
default="adam",
|
default=None,
|
||||||
help="Optimizer to use for training: adam or adamw",
|
help="Optimizer to use for training: adam or adamw",
|
||||||
)
|
)
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
|
Loading…
Reference in New Issue
Block a user