(self, low_x, high_x)
| 205 | self.init_params() |
| 206 | |
| 207 | def forward(self, low_x, high_x): |
| 208 | x = self.ftb1(low_x) |
| 209 | x = x + high_x |
| 210 | x = self.ftb2(x) |
| 211 | x = self.upsample(x) |
| 212 | |
| 213 | return x |
| 214 | |
| 215 | def init_params(self): |
| 216 | for m in self.modules(): |