From 02ac62fb1dab0964fed211014b9dfbed31a27879 Mon Sep 17 00:00:00 2001
From: neulus <softmax@kookmin.ac.kr>
Date: Mon, 29 Sep 2025 16:19:24 +0900
Subject: [PATCH] add final ckpt

---
 main.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/main.py b/main.py
index 62ba1cf..8c88887 100644
--- a/main.py
+++ b/main.py
@@ -85,4 +85,13 @@ for epoch in range(100):
             },
             f"artifact/{wandb.run.name}/checkpoint_epoch_{epoch + 1}.pt",
         )
+
+torch.save(
+    {
+        "epoch": 100,
+        "model_state_dict": model.state_dict(),
+        "optimizer_state_dict": optimizer.state_dict(),
+    },
+    f"artifact/{wandb.run.name}/checkpoint_final.pt",
+)
 wandb.finish()