Initialize in context
hskramer opened this issue · 1 comments
hskramer commented
Parameter 'resnetv20_conv0_weight' was not initialized on context cpu(0). It was only initialized on [gpu(0), gpu(1)].
Trying to get the parameter in the proper context. If I run on one gpu I receive an assertion error on the len(grads)== len(param_names) I have also received this error when running my own nets with BatchNorm.
hskramer commented
Trying to get the parameter in the proper context. If I run on one gpu I receive an assertion error on the len(grads)== len(param_names) I have also received this error when running my own nets with BatchNorm.