Saving train state of step 5

distil-whisper/events.out.tfevents.1715073503.server02.1431090.0

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

run_distillation.py

@@ -874,7 +874,7 @@ def main():
             raw_datasets["eval"] = load_dataset(
                 dataset_dict["name"],
                 dataset_dict["config"],
-                split=dataset_dict["split"],
+                split=dataset_dict["split"], 
                 cache_dir=data_args.dataset_cache_dir,
                 token=model_args.token,
                 streaming=data_args.streaming,
@@ -1595,9 +1595,8 @@ def main():
                         eval_preds = []
                         eval_labels = []
                         eval_start = time.time()
-                        print(f" vectorized_datasets[eval_split] : {len(vectorized_datasets[eval_split])}")
-
-                        validation_dataloader = DataLoader(
+                        
+                        F = DataLoader(
                             vectorized_datasets[eval_split],
                             collate_fn=data_collator,
                             batch_size=per_device_eval_batch_size,