Solved by ptrblck in post #2 
              
              
                Could you set num_workers=0 and try to run the code again? 
There might be some issues in e.g. the data loading as described here.