explain-LXMERT / lxmert /run /vqa_test.bash
WwYc's picture
Upload 61 files
08d7644 verified
raw
history blame
476 Bytes
# The name of this experiment.
name=$2
# Save logs and models under snap/vqa; make backup.
output=snap/vqa/$name
mkdir -p $output/src
cp -r src/* $output/src/
cp $0 $output/run.bash
# See Readme.md for option details.
CUDA_VISIBLE_DEVICES=$1 PYTHONPATH=$PYTHONPATH:./src \
python src/tasks/vqa.py \
--tiny --train train --valid "" \
--llayers 9 --xlayers 5 --rlayers 5 \
--batchSize 32 --optim bert --lr 5e-5 --epochs 4 \
--tqdm --output $output ${@:3}