mmpretrain/mmcls/apis
Ma Zerun e0d32df076
[Fix] Fix a bug when load checkpoints in mulit-GPUs environment. (#527)
* Fix a bug when load checkpoints in mulit-GPUs environment.

* Map the weights to CPU
2021-11-10 17:54:43 +08:00
..
__init__.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
inference.py [Fix] Fix a bug when load checkpoints in mulit-GPUs environment. (#527) 2021-11-10 17:54:43 +08:00
test.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
train.py [Fix] Remove `DistSamplerSeedHook` if use `IterBasedRunner`. (#501) 2021-10-27 17:07:33 +08:00