ptrblck:

not sure if the progress_bar is relevant, but I just ignore it


The progress bar is where it is shown that the train_loader length is equal to 1, while it should be equal to a bigger number, corresponding to the number of images of the batch.
The result of the previous code snipet is the following:
len(dataset)  1
len(loader)  1
expected number of batches 2642

I set N=42264 manually --> the number of images withing my training folder.
batch_size=16, set according to the config file