A question about HF2VAD

[tylgczq]

Thank you for your great work, I have a question:

How to add sspcab to HF2VAD，can you send the modified code？

[tylgczq]

class up(nn.Module):
def init(self, in_ch, out_ch, bilinear=False, op="none"):
super(up, self).init()
self.bilinear = bilinear
self.op = op
if self.bilinear:
self.up = nn.Sequential(nn.Upsample(scale_factor=2, mode='bilinear', align_corners=True),
nn.Conv2d(in_ch, in_ch // 2, 1), )
else:
self.up = nn.ConvTranspose2d(in_channels=in_ch, out_channels=in_ch // 2, kernel_size=3, stride=2, padding=1,
output_padding=1)
assert op in ["concat", "none"]

    if op == "concat":
        self.conv = double_conv(in_ch, out_ch)
    else:
        self.conv = double_conv(out_ch, out_ch)

whether only need to modify the last piece of code？ self.conv = double_conv(out_ch, out_ch) Add sspcab here？