samples/rnn.rb in tensor_stream-opencl-0.2.2 vs samples/rnn.rb in tensor_stream-opencl-0.2.3
- old
+ new
@@ -73,10 +73,10 @@
losses = logits_series.zip(labels_series).collect do |logits, labels|
tf.nn.sparse_softmax_cross_entropy_with_logits(logits: logits, labels: labels)
end
total_loss = tf.reduce_mean(losses)
-train_step = TensorStream::Train::AdagradOptimizer.new(0.1).minimize(total_loss)
+train_step = TensorStream::Train::AdagradOptimizer.new(0.01).minimize(total_loss)
puts "#{tf.get_default_graph.nodes.keys.size} nodes created"
zeros_state = tf.zeros([batch_size, state_size]).eval
tf.session do |sess|
sess.run(tf.global_variables_initializer)
\ No newline at end of file