def __call__(self, dict_): bs = len(dict_[list(dict_.keys())[0]]) lambda_ = compute_mixup_lambda(bs, self.alpha, self.share_lambda) dict_ = { key: mixup(value, lambda_=lambda_) for key, value in dict_.items() } return dict_
def on_batch_start(self, state): lambda_ = compute_mixup_lambda(state.batch_size, self.alpha, self.share_lambda) for key in self.mixup_keys: state.input[key] = mixup_torch(state.input[key], lambda_=lambda_)