Solved by ptrblck in post #5 
              
              
                Thanks for the update. 
Could you execute these runs and see, if the behavior changes, as a sync wouldn’t be strictly necessary, but would help us to isolate is an internal method is broken: 

add synchronizations in the train_data loop before and after each line of code
set non_blocking=False

reru…