refactor(net): 修改 DetailNode 类的默认参数

- 将 DetailNode 类的 useBlock 参数默认值从 0 改为 1
- 这个修改可能会影响网络结构的选择和初始化

两分支wtconv
This commit is contained in:
zjut 2024-11-17 16:13:59 +08:00
parent b0030fe87f
commit 231f42e924

2
net.py
View File

@ -239,7 +239,7 @@ class DepthwiseSeparableConvBlock(nn.Module):
return x
class DetailNode(nn.Module):
def __init__(self,useBlock=0):
def __init__(self,useBlock=1):
super(DetailNode, self).__init__()
if useBlock == 0:
self.theta_phi = DepthwiseSeparableConvBlock(inp=32, oup=32)