テンソルフローのLSTMのコストを計算するためのコードは次のとおりです。コードを実行すると、オプティマイザのステップには25分かかります。最小化(コスト)操作に多くの時間がかかります
cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=pred, labels=y_class))
optimizer = tf.train.RMSPropOptimizer(learning_rate=lr).minimize(cost)
どうしてですか?これは私の体重、バイアス、私のLSTMの隠れた層の数に依存していますか? 私はGTX 840Mのネットワークを訓練しています
ウェイトとバイアスの初期化の学習率とモード、およびアクティブ化関数のタイプは何ですか? –
私は重みに 'truncated_normal_initializer'を使い、バイアスには' constant_initializer'を使っています。学習率は0.001 – Anil