edits

app.py

@@ -37,9 +37,9 @@ for name, param in net_sfm.named_parameters():
 
 state2 = torch.load('fire_imagenet.pth', map_location='cpu')
 state2['net_params'] = state['net_params']
-# state2['state_dict'] = dict(state2['state_dict'], **dim_red_params_dict);
+state2['state_dict'] = dict(state2['state_dict'], **dim_red_params_dict);
 net_imagenet = fire_network.init_network(**state['net_params']).to(device)
-net_imagenet.load_state_dict(state2['state_dict']) #, strict=False)
+net_imagenet.load_state_dict(state2['state_dict'], strict=False)
 
 # ---------------------------------------
 transform = transforms.Compose([