-
Notifications
You must be signed in to change notification settings - Fork 57
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
How to train the captioning module on ground truth proposals #47
Comments
Hi adeljalalyousif To train the captioning module on ground truth proposals, run the following: # conda activate bmt
python main.py \
--procedure train_cap \
--B 32 |
Thanks for your response, but I got this error "FileNotFoundError: [Errno 2] No such file or directory './best_prop_model.pt' " : {'B': 32, ########################################################################### I need to train the captioning module on ground truth proposals without using learned proposals |
after downloading 'best_prop_model.pt' the training is work but on cpu, how to making training run on gpu, I have |
Hi Iashin, I need to train the captioning module on ground truth proposals. What should I do?
The text was updated successfully, but these errors were encountered: