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__':
import random
import gym
print("This function is called")
class NaiveSimulator(SimulatorProcess):
def _build_player(self):
return gym.make('Breakout-v0')
......
......@@ -3,13 +3,14 @@
import abc
import tensorflow as tf
import tensorflow.compat.v1 as tf
from tensorpack import ModelDesc
from tensorpack.tfutils import gradproc, optimizer, summary, varreplace
from tensorpack.tfutils.scope_utils import auto_reuse_variable_scope
from tensorpack.utils import logger
tf.disable_eager_execution()
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