From 86d16e53d70abf95e3727821aa8bcc0cffd6b686 Mon Sep 17 00:00:00 2001
From: Akemi Izuko <akemi@noway.moe>
Date: Fri, 6 Dec 2024 17:12:38 -0700
Subject: [PATCH] O1: fix epochs

---
 one_run_audit/audit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/one_run_audit/audit.py b/one_run_audit/audit.py
index e3ceee5..1083156 100644
--- 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,