def __init__(self, args): super(LotteryVAE, self).__init__(args) LotteryAE.__init__(self, args) VAE.__init__(self, args) self.pruning = args.pruning self.mu.unprunable = True self.log_var.unprunable = True
def __init__(self, encoder, decoder, synth, args, upsampler=None): DDSSynth.__init__(self, encoder, decoder, synth, args, upsampler) VAE.__init__(self)