Update cnn3d_model.py

cnn3d_model.py

@@ -300,13 +300,13 @@ def load_model():
     args = NetData(cnndata, lindata)
 
     # weight file
-    #weight_file = 'cnn3d_epoch_300.pt'
+    weight_file = 'cnn3d_epoch_300.pt'
        
     # CNN3D model
     # device = 'cuda' if torch.cuda.is_available() else 'cpu'
     device = torch.device('cpu')
     cnn3d = CNN3D(args).to(device)
-    #cnn3d.load_state_dict(torch.load(os.path.join(base_path,'weights',weight_file), map_location=device))
+    cnn3d.load_state_dict(torch.load(weight_file, map_location=device))
     cnn3d.eval()
     #print(cnn3d)