Pausing for now, train loop should now work and adding some tests

2025-04-12 12:01:13 +01:00
parent 2b38c04a57
commit be70c4e160
13 changed files with 967 additions and 58 deletions
--- a/train.py
+++ b/train.py
@@ -310,7 +310,7 @@ def main(args):
            "num_epochs", 10
        ) - 1:
            checkpoint_file = os.path.join(
-                checkpoint_path, f"checkpoint_epoch_{epoch + 1}.pth"
+                checkpoint_path, f"checkpoint_epoch_{epoch+1}.pth"
            )
            checkpoint = {
                "epoch": epoch + 1,