def main():
args = parse_test_args()
cfg = load_test_config(args)
logging.setup_logging(cfg.OUTPUT_DIR)
logger.info(args)
logger.info("Environment info:\n" + collect_env_info())
logger.info("Loaded configuration file {}".format(args.config_file))
if args.config_file:
with open(args.config_file, "r") as cf:
config_str = "\n" + cf.read()
logger.info(config_str)
logger.info("Running with config:\n{}".format(cfg))
launch_job(cfg=cfg, init_method=args.init_method, func=test)
def main():
args = parse_args()
cfg = load_config(args)
if not os.path.exists(cfg.OUTPUT_DIR):
os.makedirs(cfg.OUTPUT_DIR)
# Setup logging format.
logging.setup_logging(cfg.OUTPUT_DIR)
logger.info(args)
logger.info("Environment info:\n" + collect_env_info())
logger.info("Loaded configuration file {}".format(args.config_file))
if args.config_file:
with open(args.config_file, "r") as cf:
config_str = "\n" + cf.read()
logger.info(config_str)
logger.info("Running with config:\n{}".format(cfg))
launch_job(cfg=cfg, init_method=cfg.INIT_METHOD, func=train)