search

kelvinxu / arctic-captions

960 stars 349 forks source link

Training parameters for Flickr30k, coco dataset? #4

Closed hyunseokmin closed 9 years ago

hyunseokmin commented 9 years ago

Hello, thank you for sharing.

Shared source contains training parameters for Flickr8k dataset. Can I use the same parameters for Flickr30K and coco dataset?

Thank you ^^;

kelvinxu commented 9 years ago

The evaluate_coco.py function should give you the parameters for MS COCO. For Flickr30k, you can just try playing with the hidden state size to something like 1300. You should get something similar.

noureldien commented 8 years ago

@hyunseokmin Please, did you manage to train the model on flickr_30k? What hyper-parameters did you use, please. Thank you.