From ee456acdb86a58cae21c93f230f3acf6082f4efc Mon Sep 17 00:00:00 2001 From: Guspan Tanadi <36249910+guspan-tanadi@users.noreply.github.com> Date: Wed, 25 Oct 2023 13:03:42 +0700 Subject: [PATCH] docs: bit misspell comment DMFUnit dmfnet --- models/networks_3d/dmfnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/networks_3d/dmfnet.py b/models/networks_3d/dmfnet.py index 08185e4..4b41b5b 100644 --- a/models/networks_3d/dmfnet.py +++ b/models/networks_3d/dmfnet.py @@ -90,7 +90,7 @@ def forward(self, x): return x4 + shortcut class DMFUnit(nn.Module): - # weighred add + # weighted add def __init__(self, num_in, num_out, g=1, stride=1,norm=None,dilation=None): super(DMFUnit, self).__init__() self.weight1 = nn.Parameter(torch.ones(1)) @@ -100,7 +100,7 @@ def __init__(self, num_in, num_out, g=1, stride=1,norm=None,dilation=None): num_mid = num_in if num_in <= num_out else num_out self.conv1x1x1_in1 = Conv3d_Block(num_in, num_in // 4, kernel_size=1, stride=1, norm=norm) - self.conv1x1x1_in2 = Conv3d_Block(num_in // 4,num_mid,kernel_size=1, stride=1, norm=norm) + self.conv1x1x1_in2 = Conv3d_Block(num_in // 4, num_mid, kernel_size=1, stride=1, norm=norm) self.conv3x3x3_m1 = nn.ModuleList() if dilation == None: