O1: fix epochs

2024-12-06 17:12:38 -07:00 · 2024-12-06 17:12:38 -07:00 · 86d16e53d7
commit 86d16e53d7
parent ebfbd88332
1 changed files with 1 additions and 1 deletions
--- a/one_run_audit/audit.py
+++ b/one_run_audit/audit.py
@ -545,7 +545,7 @@ def main():
            model_init, model_trained = train_knowledge_distillation(
                teacher=teacher_trained,
                train_dl=train_dl,
-                epochs=100,
+                epochs=hp['epochs'],
                device=DEVICE,
                learning_rate=0.001,
                T=2,