training / flax /initialisation_scripts /run_tiny_2_1_init_pt.sh
supawichwac's picture
Saving train state of step 50
55f3766 verified
raw
history blame
160 Bytes
#!/usr/bin/env bash
python create_student_model_pt.py \
--teacher_checkpoint "distil-whisper/tiny-random-whisper" \
--decoder_layers 1 \
--save_dir "./"