我正在尝试使用启用了急切执行的 tensorflow 1.15 来训练模型。对于我正在使用的火车损失
train loss = mse_loss*args.lmbda + bits_per_pixel_loss
我已经将优化器定义如下
main_optimiser = tf.train.AdamOptimiser(learning_rate=1e-3)
到这里代码工作正常错误出现在优化的最小化部分
main_step = main_optimiser.minimise(train_loss, global_step=step)
我收到以下错误：
RuntimeError: `loss` passed to Optimizer.compute_gradients should be a function when eager execution is enabled.
因此，当我在函数中计算 train_loss 时，说，loss_computer在 train 函数中构建，然后使用
main_step =main_optimiser.minimise(loss_computer, global_step=step)
我收到以下错误：
AttributeError: 'NoneType' object has no attribute 'dtype'.
如何在启用急切执行的情况下训练模型？如果我需要澄清任何事情，请告诉我。非常感谢任何帮助，谢谢！