如何为GAN实现具有多个模型的tensorflow Estimator？

Question

我曾经使用tensorflow，然后找到了Keras。使用像

这样的API在Keras中训练多个模型很容易

for i in range(nb_batches):
    model1.train_on_batch(x, y)
    model2.train_on_batch(x, y)

现在我需要回到tensorflow以获得高度的灵活性。问题是张量流中没有模型，似乎Estimator类似于Keras中的模型。但是，Estimator API仅提供model_fn和参数参数，这似乎仅限于使用一个model_fn。

我想知道我们是否可以实现Estimator，以便我们可以使用以下语法：

for i in range(nb_batches):
    ...
    my_estimator.partial_fit(x, y, MODEL='gen')
    ...
    my_estimator.partial_fit(x, y, MODEL='cri')