Saving train state of step 5

distil-whisper/events.out.tfevents.1715170865.server02.1911294.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c8058e969be15e9350e524a30e5e7c7f41b457379c3f5914c0168bff7abdd853
+size 392

run_distillation.py

@@ -1647,6 +1647,7 @@ def main():
                             position=2,
                             disable=not accelerator.is_local_main_process,
                         ):
+                            print(f"type(batch) : {type(batch)}")
                             # Model forward
                             eval_metric = eval_step(batch)
                             eval_metric = accelerator.gather_for_metrics(eval_metric)