From 8e0b32f946041c132f613d04a6d1ef2d375f250d Mon Sep 17 00:00:00 2001 From: achaiah Date: Mon, 6 May 2019 23:15:35 -0500 Subject: [PATCH] fixing docs issue --- pywick/models/segmentation/testnets/transition_down.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/pywick/models/segmentation/testnets/transition_down.py b/pywick/models/segmentation/testnets/transition_down.py index 68ae735..54a2dec 100755 --- a/pywick/models/segmentation/testnets/transition_down.py +++ b/pywick/models/segmentation/testnets/transition_down.py @@ -3,10 +3,9 @@ from torch.nn import Sequential, BatchNorm2d, ReLU, Conv2d, Dropout2d, MaxPool2d import re from itertools import chain -from .utils import RichRepr -class TransitionDown(RichRepr, Sequential): +class TransitionDown(Sequential): r""" Transition Down Block as described in [FCDenseNet](https://arxiv.org/abs/1611.09326), plus compression from [DenseNet](https://arxiv.org/abs/1608.06993) @@ -40,7 +39,7 @@ def __init__(self, in_channels: int, compression: float = 1.0, dropout: float = self.add_module('pool', MaxPool2d(kernel_size=2, stride=2)) def repr_in(self, *args, **kwargs): - res = super(RichRepr, self).__repr__() + res = self.__repr__() args = filter(lambda s: len(s) > 0, map(str, args)) kwargs = (f'{k}={v}' for k, v in kwargs.items()) desc = ', '.join(chain(args, kwargs))