| from argparse import ArgumentParser | |
| from util import ConfigParser | |
| from logger import Wandb | |
| from trainer import Trainer | |
| parser = ArgumentParser( | |
| prog='Trainer implementation, using Pytorch', | |
| description='' | |
| ) | |
| if __name__ == '__main__': | |
| parser.add_argument('-p', '--config_path') | |
| args = parser.parse_args() | |
| config = ConfigParser(args.config_path).config | |
| trainer = Trainer(config) |