diff --git a/segan/models/model.py b/segan/models/model.py index 59f80c4..71e6a09 100644 --- a/segan/models/model.py +++ b/segan/models/model.py @@ -76,7 +76,7 @@ def __init__(self, opts, name='SEGAN', super(SEGAN, self).__init__(name) self.save_path = opts.save_path self.preemph = opts.preemph - self.reg_loss = getattr(F, opts.reg_loss) + # self.reg_loss = getattr(F, opts.reg_loss) if generator is None: # Build G and D self.G = Generator(1,