Pytorch Siamese Network implementation?

I tried to implement the siamese network for image classification task according to the code below: