def __init__(self, alpha=0.5, beta=1.0, gamma=0.1): super().__init__() self.global_d = GlobalDiscriminator() self.local_d = LocalDiscriminator() self.prior_d = PriorDiscriminator() self.alpha = alpha self.beta = beta self.gamma = gamma
def __init__(self, alpha=ALPHA, beta=BETA, gamma=GAMMA): super().__init__() self.alpha = alpha self.beta = beta self.gamma = gamma if alpha!=0: self.global_dim = GlobalDIM() if beta!=0: self.local_dim = LocalDIM() if gamma!=0: self.prior_mat = PriorDiscriminator()