* [Feature] Support MPS device. * Add `auto_select_device` * Add unit tests |
||
---|---|---|
.. | ||
__init__.py | ||
inference.py | ||
test.py | ||
train.py |
* [Feature] Support MPS device. * Add `auto_select_device` * Add unit tests |
||
---|---|---|
.. | ||
__init__.py | ||
inference.py | ||
test.py | ||
train.py |