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')))
