That error sounds exactly like the one on this thread. The grader apparently uses an older version of PyTorch.