Skip to content

GitLab

Unverified Commit 21603531 authored by Lutz Roeder's avatar Lutz Roeder Committed by GitHub
Browse files

Remove constants from DenseNet (#1721) (#1907)

parent 53066353
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
torchvision/models/densenet.py
View file @ 21603531
...@@ -139,8 +139,6 @@ class DenseNet(nn.Module): ...@@ -139,8 +139,6 @@ class DenseNet(nn.Module):
but slower. Default: *False*. See `"paper" <https://arxiv.org/pdf/1707.06990.pdf>`_ but slower. Default: *False*. See `"paper" <https://arxiv.org/pdf/1707.06990.pdf>`_
""" """
__constants__ = ['features']
def __init__(self, growth_rate=32, block_config=(6, 12, 24, 16), def __init__(self, growth_rate=32, block_config=(6, 12, 24, 16),
num_init_features=64, bn_size=4, drop_rate=0, num_classes=1000, memory_efficient=False): num_init_features=64, bn_size=4, drop_rate=0, num_classes=1000, memory_efficient=False):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment