(self, x)
| 217 | show_params(self) |
| 218 | |
| 219 | def forward(self, x): |
| 220 | |
| 221 | wav_lst = [] |
| 222 | out1 = self.block1(x)[1] |
| 223 | wav_lst.append(out1) |
| 224 | out2 = self.block2(out1, x)[1] |
| 225 | wav_lst.append(out2) |
| 226 | #print(out1.shape) |
| 227 | |
| 228 | return wav_lst |
| 229 | |
| 230 | def get_params(self, weight_decay=0.0): |
| 231 | # add L2 penalty |
nothing calls this directly
no outgoing calls
no test coverage detected