diff --git a/timm/models/layers/drop.py b/timm/models/layers/drop.py index ae065277..1ab1c8f5 100644 --- a/timm/models/layers/drop.py +++ b/timm/models/layers/drop.py @@ -164,3 +164,6 @@ class DropPath(nn.Module): def forward(self, x): return drop_path(x, self.drop_prob, self.training, self.scale_by_keep) + + def extra_repr(self): + return f'drop_prob={round(self.drop_prob,3):0.3f}'