diff --git a/mmdet/models/plugins/dropblock.py b/mmdet/models/plugins/dropblock.py index 7283acf39f3..bb00ade7384 100644 --- 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}')