pytorch_lightning_spells.snapmix_utils module

SnapMix Utility Functions

Reference: Shaoli-Huang/SnapMix

Functions:

get_spm(input_tensor, target, model, image_size)

pytorch_lightning_spells.snapmix_utils.get_spm(input_tensor, target, model, image_size, half=False)[source]
Parameters:

half (bool) –