帶共享網絡的Tensorflow估計器

Question

我正在構建一個新的estimator高級api的tensorflow模型。我的模特看起來像下面的截圖：

。

實際上，模型比模型更復雜，因爲模型是用來模擬遊戲操作的。分類負責決定是否適合採取行動。然後迴歸會給出有關該行爲的詳細信息。它包含CNN和RNN的組合。

但是，由於複雜性和內存消耗，不可能將分類和迴歸同時作爲兩個網絡進行訓練和運行。另外，當我創建我的估計器時，如：

# Create the Estimator 
mnist_classifier = tf.estimator.Estimator(
    model_fn=cnn_model_fn, model_dir="/tmp/mnist_convnet_model") 

我只能爲估計器提供一個模型函數。是否可以一起訓練和運行兩個估算器？

Answer 1

將您的損失函數更改爲迴歸和分類損失的線性組合。它將是一個估計值，但會有一個損失，但會有多重推論。