def __init__(self, arg_parser):
TFUtil.disable_gpu()
self.arg_parser = arg_parser
self._enable_training = True
self.train_agents = []
self.parse_args(arg_parser)
self.build_agents()
return
def __init__(self, env, arg_parser):
TFUtil.disable_gpu()
self.env = env
self.arg_parser = arg_parser
self._enable_training = True
self.train_agents = []
self.parse_args(arg_parser)
self.build_agents()
return
def __init__(self, env, arg_parser):
# shield gpu using
# os.environ["CUDA_VISIBLE_DEVICES"] = '-1'
TFUtil.disable_gpu()
# read all the arguments in the world
self.env = env
self.arg_parser = arg_parser
self._enable_training = True
self.train_agents = []
self.parse_args(arg_parser)
# set agent, model and so on, and read the arguments in args.txt
self.build_agents()
return
def __init__(self, env, arg_parser):
TFUtil.disable_gpu()
self.env = env
self.arg_parser = arg_parser
self._enable_training = True
self.train_agents = []
#added
self.max_ep = 1
self.k = 0
self.ep_done = 0
# end added
self.parse_args(arg_parser)
self.build_agents()
return