The second part of my post was a guess that this error would come next:



 ptrblck:

and check, if you would need to adapt the in_features of self.fc1 in case it’s raising a shape mismatch error afterwards.


so check the in_features of self.fc1 and set it to 4312, which is the number of features of the input activation.