If you are using different operating systems (Linux in Colab vs. Windows on your Laptop), make sure the data paths are equal, as e.g. the sorting is not equal on those systems.
If that doesn’t help, try to update Colab to 1.7.1 and check, if you are still able to train the model or if there might be a potential issue in the latest PyTorch release.