(self, x)
| 255 | |
| 256 | |
| 257 | def forward(self, x): |
| 258 | |
| 259 | x = x.permute(0,1,3,2) # [B, 2, num_frames, num_bins] |
| 260 | out = self.inp_prelu(self.inp_norm(self.inp_conv(x))) # [b, 64, num_frames, frame_size] |
| 261 | out = self.enc_dense1(out) # [b, 64, num_frames, frame_size] |
| 262 | |
| 263 | out = self.dual_transformer(out) # [b, 64, num_frames, 256] |
| 264 | out = self.output1(out) * self.output2(out) # mask [b, 64, num_frames, 256] |
| 265 | out = self.dec_dense1(out) |
| 266 | out = self.out_conv(out) |
| 267 | out = out.permute(0,1,3,2) |
| 268 | return out |
nothing calls this directly
no outgoing calls
no test coverage detected