9157432
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
""" ATIO -- All Trains in One """ from .singleTask import * __all__ = ['ATIO'] class ATIO(): def __init__(self): self.TRAIN_MAP = { 'DLF': DLF, } def getTrain(self, args): return self.TRAIN_MAP[args['model_name']](args)
