def forward()

in src/speech_reps/models/wav2vec.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

    def forward(self, x):
        with torch.no_grad():
            z = self.model.feature_extractor(x)
            if isinstance(z, tuple):
                z = z[0]
            c = self.model.feature_aggregator(z)
        return z, c