'Keep running forever when start iterating the DataLoader

Currently, I am studying the bert model. After I create the training and development data using the SSTDataset class and PyTorch's DataLoader. I tried to iterate the train_loader to access some data. This works if I do it in Google Colab but it will keep running forever if I do it in my local Jupyter Notebook.

Is there any possible reason? I have tried to reinstall my anaconda and Jupyter notebook but it still does not work. Meanwhile, I have installed related packages properly like pytorch and transformer.

These are the output from Colab and my Jupyter Notebook:

enter image description here

pytorch-dataloader


Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution Source

Related Questions