You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
MDConv class in Mixnet,bug:Expected object of device type cuda but got device type cpu for argument #2 'weight' in call to _thnn_conv_depthwise2d_forward
#23
Open
GitHubyu007 opened this issue
Jan 13, 2022
· 0 comments
SolveMixnet:MDConv class self.layers add to the network by self.layers = nn.Sequential(*self.layers【Better change the name or something】)
Otherwise an error:RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'weight' in call to _thnn_conv_depthwise2d_forward
The text was updated successfully, but these errors were encountered:
SolveMixnet:MDConv class self.layers add to the network by self.layers = nn.Sequential(*self.layers【Better change the name or something】)
Otherwise an error:RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'weight' in call to _thnn_conv_depthwise2d_forward
The text was updated successfully, but these errors were encountered: