I saw some Pytorch problems reported for sparse matrixes which are said to be fixed in 0.4.0 version but it requires migration that I would definitely prefer to stay away. As this is not my code I would be confronted with a lot of new problems during this process.