problem at RL training part

aimagelab / meshed-memory-transformer

Meshed-Memory Transformer for Image Captioning. CVPR 2020

BSD 3-Clause "New" or "Revised" License

507 stars 136 forks source link

problem at RL training part #74

Open z972778371 opened 2 years ago

z972778371 commented 2 years ago

It trains well in XE part until it Switch to RL. Does it because of the lack of my graphics card memory？ Anyone met this problem before? Please tell me how to solve it!