在使用TensorBoard过程中报错,AttributeError: type object ‘testModel‘ has no attribute ‘training‘

问题

使用tensorboard对网络可视化的过程中遇到如下报错:

AttributeError: type object ‘testModel’ has no attribute ‘training’

在使用TensorBoard过程中报错,AttributeError: type object ‘testModel‘ has no attribute ‘training‘_第1张图片

解决办法

SummaryWriter.add_graph() 之前对网络 to(device)

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
testModel = testModel().to(device)

你可能感兴趣的:(深度学习,python,pytorch)