Skip to content
Snippets Groups Projects
Commit 2f3a9ac1 authored by Neta Zmora's avatar Neta Zmora
Browse files

Merge AMC Linear layer support

parents 3f7a9408 2bc03484
No related branches found
No related tags found
No related merge requests found
......@@ -68,7 +68,7 @@ def save_checkpoint(epoch, arch, model, optimizer=None, scheduler=None,
checkpoint['dataset'] = model.dataset
if not arch:
checkpoint['arch'] = model.arch
except NameError:
except AttributeError:
pass
if optimizer is not None:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment