def __init__(self, training, denoise_model=None, **kwargs):
    super(BimodalClassifierDataGen, self).__init__(**kwargs)
    self.training = training
    if self.training:
        assert (denoise_model is not None), 'must specify denoise model in training mode!'
    self.denoise_model = denoise_model
def __init__(self, training, denoise_model=None, **kwargs):
    super(BimodalClassifierDataGen, self).__init__(**kwargs)
    self.training = training
    if self.training:
        assert (denoise_model is not None), 'must specify denoise model in training mode!'
    self.denoise_model = denoise_model