pytorch-receptive-field
pytorch-receptive-field copied to clipboard
Feature request: nn.Identity
Many layers are not supported by this lib, even if they don't affect the resulting receptive field. Please support s.t. those layers can be quickly replaced.
or add something like an unsafe
option to the receptive field computation where all layers which are not supported are automatically replaced with nn.Identity.
Thanks for your work!