forked from nagadomi/nunif
-
Notifications
You must be signed in to change notification settings - Fork 0
/
train.py
22 lines (16 loc) · 658 Bytes
/
train.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import argparse
from pprint import pprint
from nunif.addon import load_addons
from nunif.initializer import set_seed, disable_image_lib_threads
from nunif.training.trainer import create_trainer_default_parser
def main():
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
subparsers = parser.add_subparsers(help="task", required=True)
for addon in load_addons():
subparser = addon.register_train(subparsers, create_trainer_default_parser())
args = parser.parse_args()
assert (args.handler is not None)
pprint(vars(args))
args.handler(args)
if __name__ == "__main__":
main()