|
#!/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" ) |
|
|
|
|
|
|
|
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 |
|
|