Change checkpoint name

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