I have a small CNN model that works fine on my PC-CPU.
I needed more computations power, so I moved all my data and code to Datalore.
Unfortunately, when I’m initializing my model over the Datalore Sheet, my model reaches only the lines where it needs to iterate over PyTorch dataloader and then entering into a dead loop.
This means that it can’t pull image samples for training, the model gets stuck, and I must restart the kernel.
This bug does not occur when I’m using my own PC with the same code for training.
I also tried to iterate over the dataloader outside of the model, using the code:
images, labels = iter(train_dataloader).next()
and this line of code reproduces the bug described above.
I keep searching the web for a solution but didn’t find anything.
will be greatly appreciated an answer.