code/test_preangles.py
@@ -56,7 +56,8 @@ 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) # load_filtered_state_dict(model, saved_state_dict) print 'Loading data.'
