LED/main_sport_pretrain.py

import argparse
from trainer import train_sport_pretrain as sp


def parse_config():
    parser = argparse.ArgumentParser()
    parser.add_argument('--cuda', default=True)
    parser.add_argument('--learning_rate', type=float, default=1e-3,
                        help='Ignored — actual lr comes from cfg.pretrain.lr.')
    parser.add_argument('--cfg', default='soccer', help='Sport cfg name (soccer / football)')
    parser.add_argument('--gpu', type=int, default=0)
    parser.add_argument('--info', type=str, default='pretrain')
    # unused but kept for interface symmetry with the LED mains
    parser.add_argument('--train', type=int, default=1)
    return parser.parse_args()


def main(config):
    t = sp.Trainer(config)
    t.fit()


if __name__ == '__main__':
    config = parse_config()
    main(config)