Hi, you can check your version by printing torch.__version__ and report it here.
For your code sample, what are the size of the objects in the dataset? Could you replace the p.load()with just a big torch.rand(your_size) so that I can run it without your dataset?