edits

app.py

@@ -37,7 +37,7 @@ 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)