搭建网络过程遇到这个问题,检查训练过程,已经将输入数据通过
transforms.ToTensor()
转换为Tensor格式。
经检查发现是网络结构问题。
网络搭建过程中MaxPool2d()中设置return_indices=True,这意味着它将返回一个包含值和索引的元组。后面不能直接接卷积层。