samples/iris.rb in tensor_stream-opencl-0.2.2 vs samples/iris.rb in tensor_stream-opencl-0.2.3

@@ -76,12 +76,12 @@
 predict = tf.argmax(yhat, 1)
 
 # Backward propagation
 cost    = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(labels: y, logits: yhat))
 
-updates =  TensorStream::Train::GradientDescentOptimizer.new(0.01).minimize(cost)
+# updates =  TensorStream::Train::GradientDescentOptimizer.new(0.01).minimize(cost)
 # updates =  TensorStream::Train::MomentumOptimizer.new(0.01, 0.5, use_nesterov: true).minimize(cost)
-# updates =  TensorStream::Train::RMSPropOptimizer.new(0.01).minimize(cost)
+updates =  TensorStream::Train::RMSPropOptimizer.new(0.01).minimize(cost)
 
 # Run SGD
 sess = tf.session
 init = tf.global_variables_initializer
 sess.run(init)
\ No newline at end of file