c28dddb
1
2
3
4
5
6
7
8
9
10
11
12
13
datamodules = {} def register(name): def decorator(cls): datamodules[name] = cls return cls return decorator def make(name, config): dm = datamodules[name](config) return dm from . import data_module
