import argparse

import yaml
from easydict import EasyDict as edict

from trainer import SegWorker


def train(config):
    worker = SegWorker(config)
    # worker.summary()
    worker.train()


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '-c',
        '--config',
        type=str,
        default='config/default.yml',
        help='Configuration yaml file PATH.')
    opt = parser.parse_args()
    with open(opt.config, encoding='utf-8') as f:
        data_dict = yaml.load(f, Loader=yaml.FullLoader)
    cfg = edict(data_dict)
    # ClassWorker(config=cfg).summary()
    train(config=cfg)