Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Text Caption #10

Closed
abhinavkcs11 opened this issue Feb 15, 2022 · 1 comment
Closed

Text Caption #10

abhinavkcs11 opened this issue Feb 15, 2022 · 1 comment

Comments

@abhinavkcs11
Copy link

Hello,
Great work, kudos.
How to run TAP for Text Caption task? I see no instruction to run the code for the Text Caption task.

Thank you

@zyang-ur
Copy link
Contributor

Hi @abhinavkcs11,

Thank you. Please see the following example, feel free to let me know if I could better clarify anything. Thanks :)

python -m torch.distributed.launch --nproc_per_node $num_gpu tools/run.py --tasks vqa --datasets $dataset --model $model --seed $seed --config configs/vqa/$dataset/"$refine_yml".yml --save_dir save/$refine_savedir --resume_file save/$pretrain_savedir/$savename/best.ckpt training_parameters.distributed True

for example (Textcaps)

python tools/run.py --batch_size $batch_size --gpu $gpu --tasks captioning --datasets m4c_textcaps --model m4c_captioner --seed $seed --config configs/captioning/m4c_textcaps/tap_refine.yml --save_dir save/$refine_savedir --resume_file save/pretrained/textcaps_tap_base_pretrain.ckpt training_parameters.distributed True

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants