#!/usr/bin/env bash
model_ids=("roberta-base" "bert-base-cased" "t5-small" "facebook/bart-large" "facebook/bart-large-cnn")
model_architectures=("RobertaForMaskedLM" "BertForMaskedLM" "T5ForConditionalGeneration" "BartForConditionalGeneration" "BartForConditionalGeneration" )
#model_ids=("hf-internal-testing/tiny-random-roberta" "hf-internal-testing/tiny-random-bert" "hf-internal-testing/tiny-random-t5" "hf-internal-testing/tiny-random-bart")
#model_architectures=("RobertaForMaskedLM" "BertForMaskedLM" "T5ForConditionalGeneration" "BartForConditionalGeneration")

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