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