charlie880624/STDC
1
0
Fork 0
Code Issues Pull Requests Actions Packages 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 33e8357c62
commit 455d163217
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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.strides = strides
self.downsamples = downsamples self.downsamples = downsamples
self.norm_eval = norm_eval self.norm_eval = norm_eval
self.base_channels = base_channels
self.encoder = nn.ModuleList() self.encoder = nn.ModuleList()
self.decoder = nn.ModuleList() self.decoder = nn.ModuleList()