Update inference_2.py

inference_2.py

@@ -10,7 +10,7 @@ from models import image
 
 from onnx2pytorch import ConvertModel
 
-onnx_model = onnx.load('checkpoints/efficientnet.onnx')
+onnx_model = onnx.load('models/efficientnet.onnx')
 pytorch_model = ConvertModel(onnx_model)
 
 torch.manual_seed(42)
@@ -65,14 +65,14 @@ def get_args(parser):
 
 def load_img_modality_model(args):
     rgb_encoder = pytorch_model
-    ckpt = torch.load('checkpoints/model.pth', map_location = torch.device('cpu'))
+    ckpt = torch.load('models/model.pth', map_location = torch.device('cpu'))
     rgb_encoder.load_state_dict(ckpt['rgb_encoder'], strict = True)
     rgb_encoder.eval()
     return rgb_encoder
 
 def load_spec_modality_model(args):
     spec_encoder = image.RawNet(args)
-    ckpt = torch.load('checkpoints/model.pth', map_location = torch.device('cpu'))
+    ckpt = torch.load('models/model.pth', map_location = torch.device('cpu'))
     spec_encoder.load_state_dict(ckpt['spec_encoder'], strict = True)
     spec_encoder.eval()
     return spec_encoder