Skip to content

Commit 696923d

Browse files
authored
Update fineTuneCLIPMapType.py
1 parent 488942d commit 696923d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

fineTuneCLIPMapType.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,7 @@ def train_model(
205205
if os.path.exists(Path(checkpoint_path) / "CLIPMapType.pt"):
206206
os.remove(Path(checkpoint_path) / "CLIPMapType.pt")
207207
best_score = val_metrics
208-
torch.save(model.state_dict(), Path(checkpoint_path) / "CLIPMapType.pt")
208+
torch.save(model.state_dict(), "CLIPMapType.pt")
209209

210210
return history
211211

0 commit comments

Comments
 (0)