3 ms·
You should be able to make a few small changes to support "mps". In TrainingConfig set the device to "mps". The run training. In sample.py modify parse_args()
by leopoldj 2y ago
You should be able to make a few small changes to support "mps".
In TrainingConfig set the device to "mps". The run training.
In sample.py modify parse_args() and add support for mps as a possible value for the --device argument.
- antirez 2y agoThanks! I'll try. I didn't bother believing that if this was developed heavily on CUDA, it was likely going to use kernels that were missing in MPS.