Solved by user_123454321 in post #12 
              
              
                I had to do a lot of changes to make it start training… 
      for i, data in enumerate(dataloader):
        
        inps, labs = data
        inps, labs = inps.cuda(args['device']), labs.cuda(args['device'])

        inps = Variable(inps).cuda(args['device'])
        labs = Variable(labs).cuda(arg…