From 07d0c4ae963481a225ce8243f047408d96f13fab Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 24 Jun 2022 14:58:15 -0700 Subject: [PATCH] Improve repr for DropPath module --- timm/models/layers/drop.py | 3 +++ 1 file changed, 3 insertions(+) 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}'