Solved by ptrblck in post #2 
              
              
                You are currently summing all correctly predicted pixels and divide it by the batch size. To get a valid accuracy between 0 and 100% you should divide correct_train by the number of pixels in your batch. 
Try to calculate total_train as total_train += mask.nelement().