Skip to content

Commit 729d809

Browse files
committed
adam -> adadelta
1 parent 77c1825 commit 729d809

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

rnet.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,8 @@ def run():
2626

2727
rnet_model = model_rnet.R_NET(modOpts)
2828
input_tensors, loss, acc, pred_si, pred_ei = rnet_model.build_model()
29-
train_op = tf.train.AdamOptimizer(args.learning_rate).minimize(loss)
29+
#train_op = tf.train.AdamOptimizer(args.learning_rate).minimize(loss)
30+
train_op = tf.train.AdadeltaOptimizer(1.0, rho=0.95, epsilon=1e-06,).minimize(loss)
3031

3132
#saver
3233
saver = tf.train.Saver()

0 commit comments

Comments
 (0)