From fdb34db233e05c08f808196144967fa8fc32923b Mon Sep 17 00:00:00 2001 From: yongsu-bloo Date: Wed, 4 Mar 2020 19:17:04 +0900 Subject: [PATCH 1/2] resume fix: attribute name mismatch --- fairdarts/train_search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fairdarts/train_search.py b/fairdarts/train_search.py index 6358193..56c1f29 100755 --- a/fairdarts/train_search.py +++ b/fairdarts/train_search.py @@ -123,7 +123,7 @@ def main(): start_epoch = checkpoint['epoch'] dur_time = checkpoint['dur_time'] model_optimizer.load_state_dict(checkpoint['model_optimizer']) - architect.arch_optimizer.load_state_dict(checkpoint['arch_optimizer']) + architect.optimizer.load_state_dict(checkpoint['arch_optimizer']) model.restore(checkpoint['network_states']) logging.info('=> loaded checkpoint \'{}\'(epoch {})'.format(args.resume, start_epoch)) else: From 90dd256c08b42bf4ed51553204ea2d7c26a45fa3 Mon Sep 17 00:00:00 2001 From: yongsu-bloo Date: Wed, 4 Mar 2020 19:19:52 +0900 Subject: [PATCH 2/2] thop version in requirements.txt not found --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 71abe93..c001fab 100755 --- a/requirements.txt +++ b/requirements.txt @@ -7,6 +7,6 @@ Pillow==6.2.0 pyparsing==2.4.2 python-dateutil==2.8.0 six==1.12.0 -thop==0.0.31.post1910221501 +thop>=0.0.31,<0.0.32 torch==1.1.0 torchvision==0.2.1