完整报错如下:
tensorflow.python.framework.errors_impl.InvalidArgumentError: assertion failed: [predictions must be >= 0] [Condition x >= y did not hold element-wise:] [x (res_net/dense_17/BiasAdd:0) = ] [[-2.71081591][1.54978752][2.17600775]...] [y (Cast_4/x:0) = ] [0]
[[{{node assert_greater_equal/Assert/AssertGuard/else/_1/Assert}}]] [Op:__inference_train_function_9741]
检查网络的最后输出是否没有激活函数
如下:
output = self.fc(x)
# return output # 此行报错如上
return tf.nn.sigmoid(output) # 加激活函数