Closed etetteh closed 4 years ago
Thank you for your question.
I think that is because the fc-layer has been explicitly defined beyond Net(). Therefore, self.fc3 should be removed from both the definition of Net() and Net.forward(). Feel free to contact me if it does not work~
Sincerely, Yulin Wang.
Thank you for the response. It worked after following your suggestion.
Thank you for this repo. I am trying to implement ISDA with my own model on the CIFAR-10 dataset, but for some reason I am getting issues. My model is defined below :
net = Net()
batch_size = 4 feature_num = 84 num_classes = len(classes) #10 fc = Full_layer(feature_num, num_classes)
I have done every in the train function right but for some reason I am getting RuntimeError: size mismatch, m1: [4 x 10], m2: [84 x 10] at /pytorch/aten/src/TH/generic/THTensorMath.cpp:197