extra_repr for DropBlock layer (open-mmlab#6140)

* Add extra_repr for DropBlock layer * simplify with fstring
gouthamkallempudi · Sep 27, 2021 · b0cd401 · b0cd401
1 parent 3c65819
commit b0cd401
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mmdet/models/plugins/dropblock.py b/mmdet/models/plugins/dropblock.py
@@ -79,3 +79,7 @@ def _compute_gamma(self, feat_size):
         factor = (1.0 if self.iter_cnt > self.warmup_iters else self.iter_cnt /
                   self.warmup_iters)
         return gamma * factor
+
+    def extra_repr(self):
+        return (f'drop_prob={self.drop_prob}, block_size={self.block_size}, '
+                f'warmup_iters={self.warmup_iters}')