Beispiel #1
0
    def __init__(self, worker_index, config, logdir):
        logger.warning(f'Create agent {worker_index}')
        self.dconfig = utils.DotDict(config)
        self.logdir = logdir
        self.worker_index = worker_index
        self.locals = None
        self.feed_dict = None
        self.objective_vars_oid = None
        self.datasets_initialized = False

        import tensorflow as tf
        plasma.load_plasma_tensorflow_op()

        logger.warning(f'Created agent {worker_index}')
Beispiel #2
0
 def setup(self):
     logger.warning(f'Setting up agent {self.worker_index}')
     tf.reset_default_graph()
     self.locals = self._setup(self.dconfig, self.logdir)