Output dimensions of model(data) in utils.py

[karantai]

Dear Ivan, than you very much for your work!

The output of the model in utils.py in line 250:

            output = model(data) is a tensor of [20,2] (the batch size = 20).

My question is: why does the second column mean? It would make more sense if it was [20,1] resembling the predictions of the 20 images. Thank you very much in advance !

John