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)