mmpretrain/mmcls/apis/__init__.py

9 lines
275 B
Python
Raw Normal View History

2020-05-21 21:21:43 +08:00
from .inference import inference_model, init_model
from .test import multi_gpu_test, single_gpu_test
from .train import set_random_seed, train_model
__all__ = [
'set_random_seed', 'train_model', 'init_model', 'inference_model',
'multi_gpu_test', 'single_gpu_test'
]