DataParallel' object has no attribute

https://github.com/yysijie/st-gcn/issues/83

self.model.extract_feature -> self.model.module.extract_feature

pytorch 多GPU訓(xùn)練總結(jié)（DataParallel的使用）

https://blog.csdn.net/weixin_40087578/article/details/87186613

if arg.cuda:model=model.cuda() #這里將模型復(fù)制到gpu ,默認(rèn)是cuda('0')，即轉(zhuǎn)到第一個GPU 2 if len(device_id)>1:model=torch.nn.DaraParallel(model);#前提是model已經(jīng).cuda() 了

?

總結(jié)

以上是生活随笔為你收集整理的DataParallel使用的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。