def main(args):
arg_parser = common_arg_parser()
args, unknown_args = arg_parser.parse_known_args(args)
extra_args = parse_cmdline_kwargs(unknown_args)
if args.seed is not None:
tf.random.set_seed(args.seed)
if args.debug:
print(device_lib.list_local_devices())
# save the parameters
if not os.path.exists(args.output_dir):
os.makedirs(args.output_dir)
with open(os.path.join(args.output_dir, 'parameters.log'), 'w') as p_file:
json.dump(
{
'args': args.__dict__,
'unknown_args': unknown_args,
'extra_args': extra_args
}, p_file)
model = run_task(args, extra_args)
# signal completion
(Path(args.output_dir) / 'done').touch()
def main(args):
arg_parser = common_arg_parser()
args, unknown_args = arg_parser.parse_known_args(args)
extra_args = parse_cmdline_kwargs(unknown_args)
# save the parameters
if not os.path.exists(args.output_dir):
os.makedirs(args.output_dir)
with open(os.path.join(args.output_dir, 'parameters.log'), 'w') as p_file:
json.dump(
{
'args': args.__dict__,
'unknown_args': unknown_args,
'extra_args': extra_args
}, p_file)
model = train(args, extra_args)
def main(args):
arg_parser = common_arg_parser()
args, unknown_args = arg_parser.parse_known_args(args)
extra_args = parse_cmdline_kwargs(unknown_args)
model = train(args, extra_args)