1.
网络部分参数在cuda上,部分不在
问题:使用了[]装了几个卷积(不能用list)
init的最后将[]用nn.ModuleList([])封装一下
eg:
self.up_convs = nn.ModuleList(self.up_convs)
1.
网络部分参数在cuda上,部分不在
问题:使用了[]装了几个卷积(不能用list)
init的最后将[]用nn.ModuleList([])封装一下
eg:
self.up_convs = nn.ModuleList(self.up_convs)