flax/initialisation_scripts/run_large_32_2_init.sh

#!/usr/bin/env bash

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