Hi @tom that’s what I think too!
I’ll give it a try 
I’m fed up of all this .reinforce stuff !!!
Just to be a bit more clear, what I want to learn is a mapping from images to single real numbers y_pred, and those real number should be as close to the labels/class indices labels of the images as possible, as measured by loss = F.smooth_l1_loss(y_pred, labels)
Cheers