def build_gen_out_sequence(self): return Sequential( weight_scale(nn.Conv1d(self.n_filters, self.n_channels, 1), gain=calculate_gain('linear')), Reshape([[0], [1], [2], 1]), PixelShuffle2d((1, self.n_channels)), Reshape([[0], [2], [3]]), Permute([0, 2, 1]))
def build_disc_in_sequence(self): return Sequential( weight_scale(nn.Conv1d(self.n_channels, self.n_filters, 1), gain=calculate_gain('leaky_relu')), nn.LeakyReLU(0.2))
def build_gen_out_sequence(self): return Sequential( weight_scale(nn.Conv1d(self.n_filters, self.n_channels, 1), gain=calculate_gain('linear')))