I trained testing the model with the images I had on training folder. It still gives this input:
Fake Result: tensor([[ 4.7905, 13.1582]], grad_fn=)
Real Result: tensor([[ 3.6796, 10.8140]], grad_fn=)
Something is definetly off here