From c7383bf891e2ab1e9397c40c0f643a24ef4b2501 Mon Sep 17 00:00:00 2001 From: Francisco Massa Date: Mon, 13 Jan 2020 14:48:28 +0100 Subject: [PATCH] Remove constants from DenseBlock (#1727) --- torchvision/models/densenet.py | 1 - 1 file changed, 1 deletion(-) 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__()