def setup_method(self): self.graph = tf.Graph() self.sess = tf.Session(graph=self.graph) self.sess.__enter__() logger.add_output(NullOutput()) deterministic.set_seed(1) # initialize global singleton_pool for each test case from garage.sampler import singleton_pool singleton_pool.initialize(1)
def setup_class(cls): cls.reset_tf() cls.log_dir = tempfile.TemporaryDirectory() cls.prev_log_dir = snapshotter.snapshot_dir cls.prev_mode = snapshotter.snapshot_mode snapshotter.snapshot_dir = cls.log_dir.name snapshotter.snapshot_mode = 'all' logger.add_output(NullOutput())
def setup_method(self): """Setup the Session and default Graph.""" self.graph = tf.Graph() for c in self.graph.collections: self.graph.clear_collection(c) self.graph_manager = self.graph.as_default() self.graph_manager.__enter__() self.sess = tf.compat.v1.Session(graph=self.graph) self.sess_manager = self.sess.as_default() self.sess_manager.__enter__() self.sess.__enter__() logger.add_output(NullOutput()) deterministic.set_seed(1)
def setup_method(self): self.graph = tf.Graph() for c in self.graph.collections: self.graph.clear_collection(c) self.graph_manager = self.graph.as_default() self.graph_manager.__enter__() self.sess = tf.compat.v1.Session(graph=self.graph) self.sess_manager = self.sess.as_default() self.sess_manager.__enter__() self.sess.__enter__() logger.add_output(NullOutput()) deterministic.set_seed(1) # initialize global singleton_pool for each test case from metarl.sampler import singleton_pool singleton_pool.initialize(1)
def setup_method(self): logger.add_output(NullOutput())
def setUp(self): self.graph = tf.Graph() # self.sess = tf.Session(graph=self.graph) # self.sess.__enter__() logger.add_output(NullOutput())
def setUp(self): logger.add_output(NullOutput())
def setUp(self): self.graph = tf.Graph() self.sess = tf.Session(graph=self.graph) self.sess.__enter__() logger.add_output(NullOutput()) deterministic.set_seed(1)