[trainer] 0.8.20 Add explicit setting eval/train for teacher/student

TCLResearchEurope · Apr 25, 2024 · 6b8fc06 · 6b8fc06
1 parent 933fe44
commit 6b8fc06
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/examples/trainer_vision/run_finetune.py b/examples/trainer_vision/run_finetune.py
@@ -339,6 +339,8 @@ def main(config_raw: dict[str, Any], output_path: pathlib.Path) -> None:
     with open(out_decompose_config_path, "wt") as f:
         json.dump(student_decompose_config, f)
 
+    student_model.train()
+    teacher_model.eval()
     model = KdClassificationModel(
         student_model=student_model,
         teacher_model=teacher_model,

diff --git a/examples/trainer_vision/version.py b/examples/trainer_vision/version.py
@@ -1,5 +1,5 @@
 # This is parsed by setup.py, so we need to stick to str -> int parsing
-__version__ = "0.8.19"
+__version__ = "0.8.20"
 
 _ver_major = int(__version__.split(".")[0])
 _ver_minor = int(__version__.split(".")[1])