I have CNNLstm model as follows.
class CNN(nn.Module): def __init__(self): super(CNN, self).__init__() self.conv1 = nn.Sequ