Spaces:
Sleeping
Sleeping
make directory paths delimiters compatible to ubuntu
Browse files- inference.py +3 -3
inference.py
CHANGED
|
@@ -68,7 +68,7 @@ def model_summary(args):
|
|
| 68 |
def load_multimodal_model(args):
|
| 69 |
'''Load multimodal model'''
|
| 70 |
model = ETMC(args)
|
| 71 |
-
ckpt = torch.load('checkpoints
|
| 72 |
model.load_state_dict(ckpt,strict = False)
|
| 73 |
model.eval()
|
| 74 |
return model
|
|
@@ -76,14 +76,14 @@ def load_multimodal_model(args):
|
|
| 76 |
def load_img_modality_model(args):
|
| 77 |
'''Loads image modality model.'''
|
| 78 |
rgb_encoder = image.ImageEncoder(args)
|
| 79 |
-
ckpt = torch.load('checkpoints
|
| 80 |
rgb_encoder.load_state_dict(ckpt,strict = False)
|
| 81 |
rgb_encoder.eval()
|
| 82 |
return rgb_encoder
|
| 83 |
|
| 84 |
def load_spec_modality_model(args):
|
| 85 |
spec_encoder = image.RawNet(args)
|
| 86 |
-
ckpt = torch.load('checkpoints
|
| 87 |
spec_encoder.load_state_dict(ckpt,strict = False)
|
| 88 |
spec_encoder.eval()
|
| 89 |
return spec_encoder
|
|
|
|
| 68 |
def load_multimodal_model(args):
|
| 69 |
'''Load multimodal model'''
|
| 70 |
model = ETMC(args)
|
| 71 |
+
ckpt = torch.load('checkpoints/model_best.pt', map_location = torch.device('cpu'))
|
| 72 |
model.load_state_dict(ckpt,strict = False)
|
| 73 |
model.eval()
|
| 74 |
return model
|
|
|
|
| 76 |
def load_img_modality_model(args):
|
| 77 |
'''Loads image modality model.'''
|
| 78 |
rgb_encoder = image.ImageEncoder(args)
|
| 79 |
+
ckpt = torch.load('checkpoints/model_best.pt', map_location = torch.device('cpu'))
|
| 80 |
rgb_encoder.load_state_dict(ckpt,strict = False)
|
| 81 |
rgb_encoder.eval()
|
| 82 |
return rgb_encoder
|
| 83 |
|
| 84 |
def load_spec_modality_model(args):
|
| 85 |
spec_encoder = image.RawNet(args)
|
| 86 |
+
ckpt = torch.load('checkpoints/model_best.pt', map_location = torch.device('cpu'))
|
| 87 |
spec_encoder.load_state_dict(ckpt,strict = False)
|
| 88 |
spec_encoder.eval()
|
| 89 |
return spec_encoder
|