From 0c882e981e2d71f567f68197ce5e156f95ac5276 Mon Sep 17 00:00:00 2001
From: alexcbb <alexchapin@hotmail.fr>
Date: Mon, 24 Jul 2023 07:18:31 +0200
Subject: [PATCH] Change checkpoint name
---
train_sa.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train_sa.py b/train_sa.py
index f4d00dc..910afe5 100644
--- a/train_sa.py
+++ b/train_sa.py
@@ -124,7 +124,7 @@ def main():
if not global_step % cfg["training"]["checkpoint_every"]:
# Save the checkpoint of the model.
ckpt['global_step'] = global_step
- torch.save(ckpt, args.ckpt + '/ckpt.pth')
+ torch.save(ckpt, args.ckpt + '/ckpt_' + str(global_step) + '.pth')
print(f"Saved checkpoint: {args.ckpt + '/ckpt_' + str(global_step) + '.pth'}")
# We visualize some test data
--
GitLab