edits

app.py

@@ -31,7 +31,7 @@ net_sfm = fire_network.init_network(**state['net_params']).to(device)
 net_sfm.load_state_dict(state['state_dict'])
 
 state2 = torch.load('fire_imagenet.pth', map_location='cpu')
-state2['net_params']['pretrained'] = None # no need for imagenet pretrained model
+# state2['net_params']['pretrained'] = None # no need for imagenet pretrained model
 net_imagenet = fire_network.init_network(**state2['net_params']).to(device)
 net_imagenet.load_state_dict(state2['state_dict'])