Any idea about what could I try to debug this? I checked and it gets stuck exactly before looping over the DataLoader.
optimizer.zero_grad()
model.train()
for x, batch in enumerate(dataloader):
print("DEBUG") # never prints
both optimizer.zero_grad and model.train() get properly executed.