Pytorch:TypeError: apply() missing 1 required positional argument: ‘fn‘(已解决)

TypeError: apply() missing 1 required positional argument: 'fn'

Pytorch

这问题出在哪里，我的AleNet模型是没有错误的。

出错的代码：

model = summary(AlexNet, (1, 28, 28))

出错原因，没有调用GPU 或者CPU，这里需要指定一下。 

修改后：

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = AlexNet(10).to(device)
summary(model, input_size=(1, 224, 224))

到此  问题解决。

欢迎，点赞，收藏，关注