Skip to content

Commit 83da9c2

Browse files
authored
Update a3c.py for TF 0.12
1 parent 5c13b4a commit 83da9c2

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

a3c.py

+4-5
Original file line numberDiff line numberDiff line change
@@ -74,15 +74,15 @@ def log_uniform(lo, hi, rate):
7474
sess = tf.Session(config=tf.ConfigProto(log_device_placement=False,
7575
allow_soft_placement=True))
7676

77-
init = tf.initialize_all_variables()
77+
init = tf.global_variables_initializer()
7878
sess.run(init)
7979

8080
# summary for tensorboard
8181
score_input = tf.placeholder(tf.int32)
82-
tf.scalar_summary("score", score_input)
82+
tf.summary.scalar("score", score_input)
8383

84-
summary_op = tf.merge_all_summaries()
85-
summary_writer = tf.train.SummaryWriter(LOG_FILE, sess.graph_def)
84+
summary_op = tf.summary.merge_all()
85+
summary_writer = tf.summary.FileWriter(LOG_FILE, sess.graph)
8686

8787
# init or load checkpoint with saver
8888
saver = tf.train.Saver()
@@ -159,4 +159,3 @@ def signal_handler(signal, frame):
159159

160160
saver.save(sess, CHECKPOINT_DIR + '/' + 'checkpoint', global_step = global_t)
161161

162-

0 commit comments

Comments
 (0)