- Notifications
You must be signed in to change notification settings - Fork 250
Closed
Description
Hello! I am trying to finetune OFA-large on VQA using custom dataset, using the finetuning instruction in the repo. I have checked my .tsv and .pkl file several times and they are correct as your provided sample. But after command "bash train_vqa_distributed.sh", the terminal just prints:
total_num_updates 40000
warmup_updates 1000
lr 5e-5
patch_image_size 480
The GPU usage will rise to a certain value and then suddenly return to zero, and then the program will end. I train on single server with 2 GPU. Looking forward to reply, thanks for your sharing work!
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels