- 博客(1)
- 收藏
- 关注
原创 pytorch 坑
1. 网络部分参数在cuda上,部分不在 问题:使用了[]装了几个卷积(不能用list) init的最后将[]用nn.ModuleList([])封装一下 eg: self.up_convs = nn.ModuleList(self.up_convs) ...
2019-04-18 15:34:58 174 1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
1. 网络部分参数在cuda上,部分不在 问题:使用了[]装了几个卷积(不能用list) init的最后将[]用nn.ModuleList([])封装一下 eg: self.up_convs = nn.ModuleList(self.up_convs) ...
2019-04-18 15:34:58 174 1
TA创建的收藏夹 TA关注的收藏夹
TA关注的人