training / flax /initialisation_scripts /run_small_12_2_init.sh
supawichwac's picture
Saving train state of step 50
55f3766 verified
raw
history blame
196 Bytes
#!/usr/bin/env bash
TCMALLOC_LARGE_ALLOC_REPORT_THRESHOLD=10000000000 python create_student_model.py \
--teacher_checkpoint "openai/whisper-small.en" \
--decoder_layers 2 \
--save_dir "./"