From @tibuch, modifying the loss so that the masking is per pixel instead of per patch. I drop the idea here otherwise it might get forgotten.