test_stac_on_molweni.sh

export CUBLAS_WORKSPACE_CONFIG=:16:8 # to eliminate LSTM non-deterministic

if [[ "$1" == "--link_only" ]]
then
    prefix=stac_on_molweni_link_only_checkpoints
else
    prefix=stac_on_molweni_checkpoints
fi

GPU_ID=3

for seed in 1 2 3 4 5
do
  for lr in 2e-5
  do
    for epochs in 6
    do
        python main.py --encoder_model "$prefix"/"$seed"_"$lr"_"$epochs"/ \
          --seed $seed \
          --gpu $GPU_ID \
          --data_dir data/molweni \
          --max_num_test_contexts 14 \
          --eval \
          --fp16 \
          $1
    done
  done
done