for ii, (data, label) in tqdm(enumerate(train_dataloader)):
     input = Variable(data)
     target = Variable(label)
     optimizer.zero_grad()
     score = model(input) # stuck here 
     loss = criterion(score, target)
     loss.backward()
     optimizer.step()


On CPU, no multiprocessing i think…