MILVLG / bottom-up-attention.pytorch

A PyTorch reimplementation of bottom-up-attention models
Apache License 2.0
294 stars 76 forks source link

How to train my datasets #102

Open yangwenshuangshuang opened 1 year ago

yangwenshuangshuang commented 1 year ago

I am a student who is trying to understand this model. May I ask what changes do I need to make to the source code if I want to train my own dataset? Thx