Solved by jmandivarapu1 in post #21 
              
              
                Unfortunately I think I can’t help much. As the file in your github running successfully for me for different batch sizes and differ workers also. My gpu utilization is around 22%. But I am using linux machine and I do have 32 cpu’s in it. 
But I did found one interesting article about num_wprkers>0…