torch.nn.modules.activation.ReLU is not a Module subclass

torch.nn.modules.activation.ReLU is not a Module subclass

在这里插入图片描述
定位到问题代码:

在这里插入图片描述
修改为:

fc_layers.append(nn.ReLU())

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_42363032/article/details/126103803