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