几天前，他们将google colab更新为新版本的tensorflow 1.14.0-rc1，现在将tpu与keras结合使用的方式已经发生了很大变化，我再也无法使用keras模型train_on_batch函数了，只有fit函数，即使train_generator也不起作用。您知道任何解决方法吗？
注意事项：
1我试图将以前的版本重新安装到colab（1.13.1）上，我的旧代码可以正常工作，但现在的培训比以前慢得多，例如慢10到20倍，这使它成为我的选择。
2我尝试使用fit函数，就像它是train_on_batch（只有一个步骤，只有一个时期）一样，它非常慢，因为tpu训练会在函数开始时进行某种初始化几次配合就只能进行一次，这是完美的，但事实并非如此。
3我确实确实需要使用train_on_batch而不是fit函数，我正在训练需要使用该功能的GAN（发电机批处理序列，然后是鉴别器批处理序列）。
如果您想查看带有错误的示例代码： https://drive.google.com/open?id=1MDQlw3Z8v9C8SfYz_r6uJcis9odqAsE1
train_on_batch函数不应提示任何错误，但会提示错误，并且不能确切显示其含义，这似乎与急切执行有关。