Thanks a lot for the very straight forward answer! This means to get my layers as they are being shown in the torchinfo.summary(), I’d need to also get the pooling layers and not only the convolution layers from my model, right?
Thanks a lot for the very straight forward answer! This means to get my layers as they are being shown in the torchinfo.summary(), I’d need to also get the pooling layers and not only the convolution layers from my model, right?