search

marianne-m / brouhaha-vad

Predicts the level of noise and reverberation on your audiofiles
MIT License
138 stars 24 forks source link

[BUG] in einops but impact to brouhaha model forward() #20

Closed yoesak closed 5 months ago

yoesak commented 6 months ago

I found in the brouhaha-vad using the einops and cause this error "AttributeError: 'PyanNet' object has no attribute 'example_output'", the solution is changing with the torch.permute() function,

def forward(self, x: torch.Tensor): .. out = rearrange(out, "n b t o -> b t (n o)") return out `

how do I retrain the model (best.ckpt) with the new implementation?

this is the same is issue like in here

https://github.com/pyannote/pyannote-audio/issues/1620

MarvinLvn commented 5 months ago

Should be solved by https://github.com/marianne-m/brouhaha-vad/pull/21 Shout out to @ylacombe !