yiningmao's picture
Upload 34 files
c5db72e
raw
history blame contribute delete
145 Bytes
#!/bin/bash
python main.py --data_dir data/VUA20 --task_name vua --model_type MELBERT --train_batch_size 32 --learning_rate 3e-5 --warmup_epoch 2