def main():
args = parse_test_args()
cfg = load_test_config(args)
logger = logging.setup_logging(__name__, output_dir=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_train_args()
cfg = load_config(args)
logger = setup_logger("TSN", save_dir=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(args, cfg, train)