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

..

data

…

test_data

[Feature] Add CustomDataset. (#738)

2022-03-30 15:28:26 +08:00

test_downstream

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

test_metrics

[Enhance] Use PyTorch official one_hot to implement convert_to_one_hot. (#696)

2022-02-28 10:20:31 +08:00

test_models

[Feature] Support MPS device. (#894)

2022-07-28 12:28:51 +08:00

test_runtime

…

test_utils

[Feature] Support MPS device. (#894)

2022-07-28 12:28:51 +08:00