| #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 | |