def __init__()

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

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

    def __init__(self, fname, gpu):
        self.gpu = gpu
        self.model = PretrainedWav2Vec2Model(fname)
        if self.gpu is not None:
            self.model = self.model.cuda(gpu)