problem at RL training part
z972778371 opened this issue · comments
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!
aimagelab / meshed-memory-transformer
Meshed-Memory Transformer for Image Captioning. CVPR 2020