mixmo.augmentations.mixing_blocks._single_mix

mixmo.augmentations.mixing_blocks._single_mix(method, lam, input_size, config_mix)

Computes masks for two inputs (traditional MSDA methods)

method: string, mixing to use lam: float

masks: list of torch.Tensor. The masks used for mixing.

lams: list of torch.Tensor. The lams ratio on the final masks.