After defining my model:
‘’
def forward(self,x):
C1=F.relu(self.CONV1(x))
M1=self.MAX1(C1)
C2=F.relu(self.CONV2(M1))
M2=self.MAX2(C2)
C3=F.relu(self.CONV3(M2))
M3=self.MAX3(C3)
h0= Variable(torch.zeros(2,64,500))
c0=Variable(torch.zeros(2,64,500))
Pred,(hn,cn)=self.LSTM(M3,(h0,c0))''

With ‘’ summary(net, (1,4000)) ‘’ , I keep getting the following error:
’ AttributeError: ‘tuple’ object has no attribute ‘size’ ’
Does anyone know how to fix it ?
Thank you