src/speech_reps/featurize.py [144:152]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if self._ctx is not None:
            data = data.cuda(self._ctx)
        padding_mask = (
            torch.BoolTensor(data.shape).fill_(False).to(data.device)
        )
        vecs = self._model(data, padding_mask).squeeze(0)
        reps = vecs.cpu().detach().numpy()

        return reps
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/speech_reps/featurize.py [225:234]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if self._ctx is not None:
            data = data.cuda(self._ctx)
        padding_mask = (
            torch.BoolTensor(data.shape).fill_(False).to(data.device)
        )
        vecs = self._model(data, padding_mask).squeeze(0)

        reps = vecs.cpu().detach().numpy()

        return reps
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -