Feature
Desired Behavior / Functionality
Add support for mps, the cuda equivalent for mac
Instead of restating ' "cuda:0" if torch.cuda.is_available() else "cpu" ' lets have a import called device, from ... import device which in its definition handles coda/cpu/mps
What Needs to Be Done
Create device import.
Replace all statically set devices with this
How Can It Be Tested
A test should include creating a tensor, on the cpu, importing the device module and moving the tensor to the device then checking the device again.
Feature
Desired Behavior / Functionality
Add support for mps, the cuda equivalent for mac
Instead of restating ' "cuda:0" if torch.cuda.is_available() else "cpu" ' lets have a import called device,
from ... import devicewhich in its definition handles coda/cpu/mpsWhat Needs to Be Done
Create device import.
Replace all statically set devices with this
How Can It Be Tested
A test should include creating a tensor, on the cpu, importing the device module and moving the tensor to the device then checking the device again.