Skip to content

Commit

Permalink
examples: comparison w/ mars optimizing 1d
Browse files Browse the repository at this point in the history
  • Loading branch information
stockeh committed Nov 21, 2024
1 parent 87d7996 commit e1b49cd
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions examples/networks/compare.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,8 +60,6 @@ def get_optimizers(args):
"learning_rate": get_cosine_schedule(3e-3, 1e-6, n_warmup, decay_steps),
"weight_decay": weight_decay,
"optimize_1d": True,
# "learning_rate_1d": get_cosine_schedule(3e-3, 1e-6, n_warmup, decay_steps),
# "weight_decay_1d": weight_decay,
},
),
]
Expand Down

0 comments on commit e1b49cd

Please sign in to comment.