Commit 39bf1e59 authored by Shashank Suhas's avatar Shashank Suhas

Changed tf compatibility and disabled eager execution

parent fcfc899d
...@@ -160,6 +160,8 @@ if __name__ == '__main__': ...@@ -160,6 +160,8 @@ if __name__ == '__main__':
import random import random
import gym import gym
print("This function is called")
class NaiveSimulator(SimulatorProcess): class NaiveSimulator(SimulatorProcess):
def _build_player(self): def _build_player(self):
return gym.make('Breakout-v0') return gym.make('Breakout-v0')
......
...@@ -3,13 +3,14 @@ ...@@ -3,13 +3,14 @@
import abc import abc
import tensorflow as tf import tensorflow.compat.v1 as tf
from tensorpack import ModelDesc from tensorpack import ModelDesc
from tensorpack.tfutils import gradproc, optimizer, summary, varreplace from tensorpack.tfutils import gradproc, optimizer, summary, varreplace
from tensorpack.tfutils.scope_utils import auto_reuse_variable_scope from tensorpack.tfutils.scope_utils import auto_reuse_variable_scope
from tensorpack.utils import logger from tensorpack.utils import logger
tf.disable_eager_execution()
class Model(ModelDesc): class Model(ModelDesc):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment