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}')
def setup(self): logger.warning(f'Setting up agent {self.worker_index}') tf.reset_default_graph() self.locals = self._setup(self.dconfig, self.logdir)