src/speech_reps/models/wav2vec.py [57:64]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if self.gpu is not None:
            self.model = self.model.cuda(gpu)

    def __call__(self, x):
        x = torch.from_numpy(x).float()
        if self.gpu is not None:
            x = x.cuda(self.gpu)
        with torch.no_grad():
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/speech_reps/models/wav2vec2.py [55:62]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if self.gpu is not None:
            self.model = self.model.cuda(gpu)

    def __call__(self, x):
        x = torch.from_numpy(x).float()
        if self.gpu is not None:
            x = x.cuda(self.gpu)
        with torch.no_grad():
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -