ruotianluo / self-critical.pytorch

Unofficial pytorch implementation for Self-critical Sequence Training for Image Captioning. and others.
MIT License
991 stars 278 forks source link

损失函数迁移 #285

Open Jack2Lu opened 5 months ago

Jack2Lu commented 5 months ago

你好,谢谢你的分享,我看到很多篇论文中rl损失相对于普通的ce损失有所提点,想把self-critical损失函数迁移到其他框架当中,请问如何做到呢