From 9f2fa4a5fa2af40503c05379c9d5318e87581c7b Mon Sep 17 00:00:00 2001 From: natanielruiz <nataniel777@hotmail.com> Date: 星期五, 08 十二月 2017 02:10:14 +0800 Subject: [PATCH] Removed shape loading which fixes a bug with loading AFLW2000 and 300W-LP for training and testing. --- code/test_resnet50_regression.py | 9 +-------- 1 files changed, 1 insertions(+), 8 deletions(-) diff --git a/code/test_resnet50_regression.py b/code/test_resnet50_regression.py index 6945269..67c63af 100644 --- a/code/test_resnet50_regression.py +++ b/code/test_resnet50_regression.py @@ -36,13 +36,6 @@ return args -def load_filtered_state_dict(model, snapshot): - # By user apaszke from discuss.pytorch.org - model_dict = model.state_dict() - snapshot = {k: v for k, v in snapshot.items() if k in model_dict} - model_dict.update(snapshot) - model.load_state_dict(model_dict) - if __name__ == '__main__': args = parse_args() @@ -55,7 +48,7 @@ print 'Loading snapshot.' # Load snapshot saved_state_dict = torch.load(snapshot_path) - load_filtered_state_dict(model, saved_state_dict) + model.load_state_dict(saved_state_dict) print 'Loading data.' -- Gitblit v1.8.0