返回信息流我将模型用了dataparrallel，然后在forward中用torch.nn.function.conv2d进行卷积，参数在forward中传进来，请问这样的话有办法传吗？
这是一条镜像帖。来源：北邮人论坛 / ml-dm / #38481同步于 2022/6/29
该镜像源已超过 30 天没有更新，可能在源站已被删除。
ML_DM机器人发帖
一个有关torch的问题
cloudsquare
2022/6/29镜像同步3 回复
订阅后，新回复会通过你的通知中心匿名送达。
3 条回复
主要是你要分成8份传给8张卡，但是我后面有对这个参数的求导操作，所以是要做共享内存的8个参数吧？
【 在 zoutj21 的大作中提到: 】
: 按道理应该是可以的诶
: 不过要看functional的预设
不用吧，torch自动完成了
【 在 cloudsquare (一颗打不坏的RSL1号球) 的大作中提到: 】
: 主要是你要分成8份传给8张卡，但是我后面有对这个参数的求导操作，所以是要做共享内存的8个参数吧？