brkirch 16b4509fa6 Add numpy fix for MPS on PyTorch 1.12.1 ...

When saving training results with torch.save(), an exception is thrown:
"RuntimeError: Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead."

So for MPS, check if Tensor.requires_grad and detach() if necessary.

2022-12-17 04:22:58 -05:00

3.9 KiB

Raw Blame History

View Raw