Closed BPdeRooij closed 10 months ago
Fixes #{9} by transforming mpp from float64 to torch.float32 if mps backend is available. Also adds an explicit trainer/mps.yaml config file, based on the trainer/default.yaml
mpp
float64
torch.float32
mps
trainer/mps.yaml
trainer/default.yaml
Why is mpp even in there?
Fixes #{9} by transforming
mpp
fromfloat64
totorch.float32
ifmps
backend is available. Also adds an explicittrainer/mps.yaml
config file, based on thetrainer/default.yaml