From 455d1632171ef8c2ce7bb508cdf2f4e9ea14f270 Mon Sep 17 00:00:00 2001
From: lizz <innerlee@users.noreply.github.com>
Date: Wed, 31 Mar 2021 00:52:08 +0800
Subject: [PATCH] Save base_channels for UNet (#450)

Signed-off-by: lizz <lizz@sensetime.com>
---
 mmseg/models/backbones/unet.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mmseg/models/backbones/unet.py b/mmseg/models/backbones/unet.py
index 1172937..6cbda00 100644
--- a/mmseg/models/backbones/unet.py
+++ b/mmseg/models/backbones/unet.py
@@ -329,6 +329,7 @@ class UNet(nn.Module):
         self.strides = strides
         self.downsamples = downsamples
         self.norm_eval = norm_eval
+        self.base_channels = base_channels
 
         self.encoder = nn.ModuleList()
         self.decoder = nn.ModuleList()