Commit 002e973d authored by Anthony Larcher's avatar Anthony Larcher
Browse files


parent 00ec1954
......@@ -1191,7 +1191,7 @@ def get_optimizer(model, model_opts, train_opts, training_loader):
if train_opts["scheduler"]["type"] == 'CyclicLR':
cycle_momentum = True
if train_opts["optimizer"]["type"] in ["aam", "aps"]:
if train_opts["optimizer"]["type"] == "adam":
cycle_momentum = False
scheduler = torch.optim.lr_scheduler.CyclicLR(optimizer=optimizer,
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment