mirrors
/
mmclassification
mirror of https://github.com/open-mmlab/mmclassification.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmclassification/tools
History
Ma Zerun c03efeeea4
[Feature] Support MPS device. (#894) 
* [Feature] Support MPS device.

* Add `auto_select_device`

* Add unit tests
 		2022-07-28 12:28:51 +08:00
..
analysis_tools [Refactor] Remove deprecation. (#633) 2022-02-10 09:30:18 +08:00
convert_models [Feature] Support VAN. (#739) 2022-04-28 09:35:17 +08:00
deployment [Enhance] Return code 1 if scripts runs failed. (#694) 2022-03-03 10:54:16 +08:00
misc [Refactor] Remove deprecation. (#633) 2022-02-10 09:30:18 +08:00
visualizations [Enhance] Enhance vis-pipeline tool. (#604) 2022-03-04 14:40:02 +08:00
dist_test.sh [Feature] Add multi machine dist_train (#734) 2022-03-23 12:03:03 +08:00
dist_train.sh [Feature] Add multi machine dist_train (#734) 2022-03-23 12:03:03 +08:00
kfold-cross-valid.py [Fix] Fix wrong `--local_rank`. 2022-05-17 21:10:25 +08:00
slurm_test.sh
…
slurm_train.sh
…
test.py [Feature] Support MPS device. (#894) 2022-07-28 12:28:51 +08:00
train.py [Feature] Support MPS device. (#894) 2022-07-28 12:28:51 +08:00