Mat1 and mat2 shapes cannot be multiplied!

[GHALI_Ahmed]

I want to thank you very much!
via your posts that you’re include in the last reponse, i found this beautiful post:

How can I replace the forward method of a predefined torchvision model with my customized forward function?

An alternative to @ptrblck 's solution is this: import torchvision.models as models import torch def new_forward(self, x): x = self.conv1(x) x = self.bn1(x) x = self.relu(x) x = self.maxpool(x) x = self.layer1(x) x = self.layer2(x) x = self.layer3(x) x = self.layer4(x) return x # define a resnet instance resent = models.resnet18() # add new_forward function to the resnet instance as a class method bound_method = new_forward.__get__(resent, resent.__clas…

have a nice day