Could you print the device
attribute of all parameters before calling train
?
for name, param in model.named_parameters():
device = param.device
if device == torch.device('cpu'):
print('ERROR in', name, device)
Could you print the device
attribute of all parameters before calling train
?
for name, param in model.named_parameters():
device = param.device
if device == torch.device('cpu'):
print('ERROR in', name, device)