training / flax /initialisation_scripts /run_large_32_2_init.sh

Saving train state of step 50

55f3766 verified over 2 years ago

195 Bytes

#!/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 "./"

	#!/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 "./"