def __init__(self, args, worker_proto, network_proto):
self.args = args
self._npr = np.random.RandomState(args.seed + 23333)
self._observation_size, self._action_size, \
self._action_distribution = \
env_register.io_information(self.args.task)
self._worker_type = worker_proto
self._network_type = network_proto
# init the multiprocess actors
self._task_queue = multiprocessing.JoinableQueue()
self._result_queue = multiprocessing.Queue()
self._init_workers()
self._build_env()
self._base_path = init_path.get_abs_base_dir()
self._current_iteration = 0
def _set_io_size(self):
self._observation_size, self._action_size, \
self._action_distribution = \
env_register.io_information(self.args.task)