def run(self):
with distributed_utils.slurm_distributed_context(self.opt) as opt:
self.train_loop = single_train.TrainLoop(opt)
self.parser = self.parser
self.parser.opt = self.train_loop.agent.opt
self.parser.print_args()
return self.train_loop.train()
def main():
parser = eval_model.setup_args()
parser.add_distributed_training_args()
parser.add_argument('--port', type=int, default=61337, help='TCP port number')
opt = parser.parse_args(print_args=(os.environ['SLURM_PROCID'] == '0'))
with distributed_utils.slurm_distributed_context(opt) as opt:
return eval_model.eval_model(opt)
def run(self):
with distributed_utils.slurm_distributed_context(self.opt) as opt:
return TodWorldScript(opt).run()
def run(self):
with distributed_utils.slurm_distributed_context(self.opt) as opt:
self.train_loop = single_train.TrainLoop(opt)
return self.train_loop.train()
def run(self):
with distributed_utils.slurm_distributed_context(self.opt) as opt:
return eval_model.eval_model(opt)
