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