Spaces:

naver
/

SuperFeatures

Build error

YannisK commited on Apr 21, 2022

Commit

ce415a7

•

1 Parent(s): 3333ebe

edits

Files changed (1) hide show

app.py CHANGED Viewed

@@ -31,8 +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
-net_imagenet = fire_network.init_network(**state2['net_params']).to(device)
 net_imagenet.load_state_dict(state2['state_dict'])
 # ---------------------------------------

 net_sfm.load_state_dict(state['state_dict'])
 state2 = torch.load('fire_imagenet.pth', map_location='cpu')
+net_imagenet = fire_network.init_network(**state['net_params']).to(device)
 net_imagenet.load_state_dict(state2['state_dict'])
 # ---------------------------------------