File size: 755 Bytes
f0d385b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
#!/usr/bin/env bash
#model_ids=("hf-internal-testing/tiny-random-roberta" "hf-internal-testing/tiny-random-bert" "hf-internal-testing/tiny-random-bart" "tf-internal-testing/tiny-random-t5")
#model_architectures=("RobertaForMaskedLM" "BertForMaskedLM" "BartForConditionalGeneration" "T5ForConditionalGeneration")
model_ids=("hf-internal-testing/tiny-random-roberta")
model_architectures=("RobertaForMaskedLM")
rm -rf log.txt
touch log.txt
for model_idx in "${!model_ids[@]}"; do
model_id=${model_ids[model_idx]}
model_architecture=${model_architectures[model_idx]}
echo "Check ${model_id} ..." >> log.txt
./check_gradients_pt_flax.py "${model_id}" "${model_architecture}" >> log.txt
echo "=========================================" >> log.txt
done
|