repeat v1 compat

examples/A3C-Gym/train-atari.py

@@ -11,7 +11,7 @@ import numpy as np
 import os
 import sys
 import uuid
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 from six.moves import queue
 from concurrent import futures
 CancelledError = futures.CancelledError
@@ -27,6 +27,8 @@ from atari_wrapper import FireResetEnv, FrameStack, LimitLength, MapState
 from common import Evaluator, eval_model_multithread, play_n_episodes
 from simulator import SimulatorMaster, SimulatorProcess, TransitionExperience
 
+tf.disable_eager_execution()
+
 IMAGE_SIZE = (84, 84)
 FRAME_HISTORY = 4
 GAMMA = 0.99
@@ -169,6 +171,7 @@ class MySimulatorMaster(SimulatorMaster, Callback):
         """
         Launch forward prediction for the new state given by some client.
         """
+        print(state)
         def cb(outputs):
             try:
                 distrib, value = outputs.result()