Closed EnricoTrizio closed 4 months ago
The problem should have been fixed in PyTorch from version 2.1 in https://github.com/pytorch/pytorch/pull/111576 The test env installs PyTorch from conda-forge. Probably doing it from the PyTorch channel would give the latest pytorch where the device issue has been solved
I tried to use the latest version of PyTorch, as they apparently addressed the problem, but it didn't change much for us. For now, I would test it on MacOS-13 and leave an open issue to remember to change it in the future when the 14 becomes stable again.
Description
Tests failing:
I added the function explicitly in the datamodule file
This closes #128
TODO
Check if tests succeed
Status