Import evaluate_passkey_retrieval for use in the train methond

jshuadvd · Jul 5, 2024 · 7f2f11c · 7f2f11c
1 parent b0a7f9d
commit 7f2f11c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/train.py b/train.py
@@ -17,6 +17,8 @@
 import wandb
 import os
 
+from evaluation import evaluate_passkey_retrieval
+
 reload(src.main)
 
 # Initialize the accelerator