santi-pdp / segan_pytorch

Speech Enhancement Generative Adversarial Network in PyTorch
MIT License
379 stars 109 forks source link

OutGate is missing #25

Open develooper1994 opened 4 years ago

develooper1994 commented 4 years ago

generator.py class Generator1D(Model): ... if post_proc: self.comb_net = PostProcessingCombNet(1, 512) if out_gate: self.out_gate = OutGate(1, 1) <- There isn't any OutGate if big_out_filter: self.out_filter = nn.Conv1d(1, 1, 513, padding=513 // 2) ...