예제 #1
0
    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)
예제 #2
0
    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())
예제 #3
0
 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)
예제 #4
0
    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)
예제 #5
0
 def setup_method(self):
     logger.add_output(NullOutput())
예제 #6
0
 def setUp(self):
     self.graph = tf.Graph()
     # self.sess = tf.Session(graph=self.graph)
     # self.sess.__enter__()
     logger.add_output(NullOutput())
예제 #7
0
 def setUp(self):
     logger.add_output(NullOutput())
예제 #8
0
 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)