mmpretrain/mmcls/apis
WRH b99bd4fa88
Fix bug for CPU training (#286)
* remove MMDataParallel when using cpu

* support cpu testing

* fix lint
2021-06-12 22:26:33 +08:00
..
__init__.py Visualize results on image demo (#58) 2020-10-10 16:33:27 +08:00
inference.py [Fix] check for CLASSES in checkpoint meta (#207) 2021-04-15 22:19:23 +08:00
test.py [Bug] Missing test data when num_imgs can not be evenly divided by num_gpus (#299) 2021-06-11 15:53:23 +08:00
train.py Fix bug for CPU training (#286) 2021-06-12 22:26:33 +08:00