Save base_channels for UNet (#450)

Signed-off-by: lizz <lizz@sensetime.com>
open-mmlab · Mar 30, 2021 · 455d163 · 455d163
1 parent 33e8357
commit 455d163
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmseg/models/backbones/unet.py b/mmseg/models/backbones/unet.py
@@ -329,6 +329,7 @@ def __init__(self,
         self.strides = strides
         self.downsamples = downsamples
         self.norm_eval = norm_eval
+        self.base_channels = base_channels
 
         self.encoder = nn.ModuleList()
         self.decoder = nn.ModuleList()