yangbang18 / Non-Autoregressive-Video-Captioning

The PyTorch code of the AAAI2021 paper "Non-Autoregressive Coarse-to-Fine Video Captioning".
58 stars 6 forks source link

作者您好,请问这个代码可以多GPU运行吗? #6

Closed LiMxStar closed 3 years ago

yangbang18 commented 3 years ago

在已提取视频特征的情况下,训练所需GPU资源较少,所以目前只在单卡上运行。若想要多GPU运行,你可以参考知乎链接官方文档,我相信需要改动的代码并不多。