File size: 319 Bytes
d73500e | 1 2 3 4 5 6 7 8 9 10 11 | #!/usr/bin/bash
model_path="########PATH_TO_HUGGING_FACE_CHECKPOINT########"
save_file="########PATH_TO_SAVE_THE_RESULTS########/speed.csv"
model_type="normal" # normal or quantized
python src/benchmark_speed.py \
--model_path $model_path \
--model_type ${model_type} \
--save_file ${save_file} \
--pretrained |