From 1c76007da710ffa15d32756f3528bf78c5f3ef49 Mon Sep 17 00:00:00 2001 From: "kangxiaoyang.kxy" Date: Sat, 19 Aug 2023 16:27:09 +0800 Subject: [PATCH] update image colorization model: remove redundant layers for less memory use Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13669766 * update image colorization model: remove redundant layers for less memory use --- .../cv/image_colorization/ddcolor/utils/convnext.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/modelscope/models/cv/image_colorization/ddcolor/utils/convnext.py b/modelscope/models/cv/image_colorization/ddcolor/utils/convnext.py index 3da14c48..9ea14173 100644 --- a/modelscope/models/cv/image_colorization/ddcolor/utils/convnext.py +++ b/modelscope/models/cv/image_colorization/ddcolor/utils/convnext.py @@ -116,11 +116,11 @@ class ConvNeXt(nn.Module): self.add_module(layer_name, layer) self.norm = nn.LayerNorm(dims[-1], eps=1e-6) # final norm layer - self.head_cls = nn.Linear(dims[-1], 4) + # self.head_cls = nn.Linear(dims[-1], 4) self.apply(self._init_weights) - self.head_cls.weight.data.mul_(head_init_scale) - self.head_cls.bias.data.mul_(head_init_scale) + # self.head_cls.weight.data.mul_(head_init_scale) + # self.head_cls.bias.data.mul_(head_init_scale) def _init_weights(self, m): if isinstance(m, (nn.Conv2d, nn.Linear)): @@ -141,7 +141,7 @@ class ConvNeXt(nn.Module): def forward(self, x): x = self.forward_features(x) - x = self.head_cls(x) + # x = self.head_cls(x) return x