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

Training Detail #5

Closed
SmallYour opened this issue May 16, 2021 · 3 comments
Closed

Training Detail #5

SmallYour opened this issue May 16, 2021 · 3 comments

Comments

@SmallYour
Copy link

Hi, I am trying to reproduce the baseline(Topdown in Flickr 30k, with CIDEr optimization in RL phase) in your paper, and I used different kinds of code including yours, but I failed, The result of CIDEr score is around 60, can not reach the similar result in your paper (over 67), can you share the detailed training setting?

Thanks!

@YuanEZhou
Copy link
Owner

Strange! I will check the training setting later and give you feedback within this week.

@YuanEZhou
Copy link
Owner

Sorry for the later reply. I am running the code and the CIDEr score is around 64 after 60 epochs and I think it will reach 67 after 110 epochs.
捕获

To reproduce the baseline (Topdown in Flickr 30k, with CIDEr optimization in RL phase), please try:
python train.py --id CE --caption_model topdown --input_json data/flickrtalk.json --input_fc_dir data/flickrbu/flickrbu_fc --input_att_dir data/flickrbu/flickrbu_att --input_box_dir data/flickrbu/flickrbu_box --input_label_h5 data/flickrtalk_label.h5 --batch_size 29 --learning_rate 5e-4 --learning_rate_decay_start 0 --scheduled_sampling_start 0 --checkpoint_path log/CE --save_checkpoint_every 1000 --val_images_use -1 --max_epochs 30
and then
python train.py --id sc-cider-CE --caption_model topdown --input_json data/flickrtalk.json --input_fc_dir data/flickrbu/flickrbu_fc --input_att_dir data/flickrbu/flickrbu_att --input_box_dir data/flickrbu/flickrbu_box --input_label_h5 data/flickrtalk_label.h5 --batch_size 29 --learning_rate 5e-5 --start_from log/CE --checkpoint_path log/sc-cider-CE --save_checkpoint_every 1000 --language_eval 1 --val_images_use -1 --self_critical_after 30 --max_epochs 110 --cider_reward_weight 1 --ground_reward_weight -1

@SmallYour
Copy link
Author

According to your code, we successfully reproduce the performance in your paper. Thanks for replying.

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