mmpretrain

mirror of https://github.com/open-mmlab/mmpretrain.git

Author	SHA1	Message	Date
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