Closed CherishineNi closed 5 years ago
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
AttributeError: 'module' object has no attribute 'device'
AttributeError: 'module' object has no attribute 'device'