3
私は埋め込みレイヤーを持つモデルを作成しました.LSTMはgpu:0に、softmaxレイヤーはgpu:1に作成しました。TensorFlowのグラジエントを特定のGPUに固定するにはどうすればよいですか?
そうように私は、トレーニングOPを作成:
optim = tf.train.GradientDescentOptimizer(LEARNING_RATE)
train_op = optim.minimize(final_loss)
トレーニングはメモリ不足エラーで失敗します。デバイスの配置ロギングを有効にすると、softmaxの勾配がgpu:1ではなくgpu:0に設定されていることがわかります。 TensorFlowを正しく配置するにはどうすればいいですか?
ありがとう、私はドキュメントのそのオプションを見落としました。 – Maarten