Method mask_sequence

astroclip/models/specformer.py:115–117

Mask batched sequence

(self, x: Tensor)

        return loss

    def mask_sequence(self, x: Tensor):
        """Mask batched sequence"""
        return torch.stack([self._mask_seq(el) for el in x])

    def preprocess(self, x):
        std, mean = x.std(1, keepdim=True).clip_(0.2), x.mean(1, keepdim=True)