flax/initialisation_scripts/run_medium_24_2_init.sh

#!/usr/bin/env bash

TCMALLOC_LARGE_ALLOC_REPORT_THRESHOLD=10000000000 python create_student_model.py \
  --teacher_checkpoint "openai/whisper-medium.en" \
  --decoder_layers 2 \
  --save_dir "./"