(self, x)
| 141 | nn.init.constant_(m.bias.data, 0.01) |
| 142 | |
| 143 | def forward(self, x): |
| 144 | |
| 145 | x = self.quant(x) |
| 146 | res_conv0_0 = self.conv0_0(x) |
| 147 | res_conv0_1 = self.conv0_1(x) |
| 148 | res_conv0_2 = self.conv0_2(x) |
| 149 | res_conv0_3 = self.conv0_3(x) |
| 150 | |
| 151 | res = self.cat1.cat((res_conv0_0, res_conv0_1, res_conv0_2, res_conv0_3), dim=1) |
| 152 | |
| 153 | res = self.conv2(res) |
| 154 | res = self.Gblocks(res) |
| 155 | |
| 156 | res_conv1 = self.conv1(x) |
| 157 | res = self.cat2.cat((res, res_conv1), dim=1) |
| 158 | |
| 159 | res = self.conv3(res) |
| 160 | res = self.conv4(res) |
| 161 | |
| 162 | res = self.depth2spcae(res) |
| 163 | res = self.clippedReLU(res) |
| 164 | |
| 165 | res = self.dequant(res) |
| 166 | return res |
| 167 | |
| 168 | def fuse_model(self): |
| 169 | for m in self.modules(): |
nothing calls this directly
no outgoing calls
no test coverage detected