Adding new model

2023-03-12 19:39:55 +00:00
parent ba44762eba
commit ed0385d1c5
6 changed files with 18 additions and 6 deletions
--- a/src/train.py
+++ b/src/train.py
@@ -114,4 +114,6 @@ def train():
    
    print(f"Best val acc: {top_val_acc} | Best train acc: {top_train_acc} | Epoch: {checkpoint_index}")

-train()
+# Path: src/train.py
+if __name__ == "__main__":
+    train()