2017-09-11 3 views
0

テンソルフローを使用して、事前に訓練されたモデルを提供しています。 奇妙なことは、私がこのモデルのために同じデータを入力したときに、毎回異なる結果が得られたことです。テンソル流通、異なる結果を得る

変数の初期化時に問題が発生する可能性があると私は考えていました。

答えて

0

二つの一般的な問題:

  1. 変数がランダムに再初期化されるmain_opの既知の問題があります。
  2. 予測グラフにドロップアウトレイヤーを残しました。

(1)、代わりにこれを使用する対処するために:

def main_op(): 
    init_local = variables.local_variables_initializer() 
    init_tables = lookup_ops.tables_initializer() 
    return control_flow_ops.group(init_local, init_tables) 

対処するために(2)、あなたが直接あなたのトレーニングのグラフをエクスポートしていないことを確認してください。予測/提供のための新しいグラフを作成する必要があります。 tf.estimatorフレームワークを使用している場合は、modetf.estimator.ModeKeys.TRAINの場合、条件付きでドロップアウトレイヤーを追加します。