Hi. Thank you for this awesome repo. In the inference mode I don't have any problem. But when it comes to training the architecture of the model has the following problem.
RuntimeError: mat1 and mat2 shapes cannot be multiplied (2x100352 and 25088x512)
This error happens in
File "/home/marefat/projects/MagFace/run/../models/iresnet.py", line 145, in forward x = self.fc(x)
I tried to change the following:
self.fc = nn.Linear(512 * block.expansion * self.fc_scale, num_classes)
to fix this problem and match the dimensions. But after this I face core dump problem.
Any recommendation will be appreciated.
Hi. Thank you for this awesome repo. In the inference mode I don't have any problem. But when it comes to training the architecture of the model has the following problem.
RuntimeError: mat1 and mat2 shapes cannot be multiplied (2x100352 and 25088x512)This error happens in
File "/home/marefat/projects/MagFace/run/../models/iresnet.py", line 145, in forward x = self.fc(x)I tried to change the following:
self.fc = nn.Linear(512 * block.expansion * self.fc_scale, num_classes)to fix this problem and match the dimensions. But after this I face core dump problem.
Any recommendation will be appreciated.