distil-ast-audioset / eval_teacher.sh
w11wo's picture
Added eval script
03cf889
raw
history blame contribute delete
432 Bytes
python eval.py \
--model_name_or_path MIT/ast-finetuned-audioset-10-10-0.4593 \
--dataset_name bookbot/audioset \
--output_dir ast-audioset-test \
--overwrite_output_dir \
--remove_unused_columns False \
--freeze_feature_encoder False \
--do_eval \
--fp16 \
--attention_mask False \
--per_device_eval_batch_size 32 \
--dataloader_num_workers 4 \
--seed 0 \
--report_to tensorboard