rnn.rb in tensor_stream-opencl-0.2.1

- old
+ new

@@ -6,11 +6,11 @@
 #
 
 require "bundler/setup"
 require 'tensor_stream'
 require 'tensor_stream/opencl'
-require 'pry-byebug'
+# require 'pry-byebug'
 
 
 tf = TensorStream
 
 num_epochs = 100
@@ -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.3).minimize(total_loss)
+train_step = TensorStream::Train::AdagradOptimizer.new(0.1).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