mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Save base_channels for UNet (#450)

Browse Source 
Signed-off-by: lizz <lizz@sensetime.com>
This commit is contained in:
lizz 2021-03-31 00:52:08 +08:00 committed by GitHub
parent bbb9f074f6
commit fcad6df7a6
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mmseg/models/backbones/unet.py
View File

@ -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()