diff --git a/torchvision/models/densenet.py b/torchvision/models/densenet.py index d4dc283a5f5..b58e8b413e1 100644 --- a/torchvision/models/densenet.py +++ b/torchvision/models/densenet.py @@ -92,7 +92,6 @@ def forward(self, input): # noqa: F811 class _DenseBlock(nn.ModuleDict): _version = 2 - __constants__ = ['layers'] def __init__(self, num_layers, num_input_features, bn_size, growth_rate, drop_rate, memory_efficient=False): super(_DenseBlock, self).__init__()