marianne-m/brouhaha-vad

Wrong model specifications

hbredin opened this issue · 1 comments

model = Model.from_pretrained("models/best/checkpoints/best.ckpt")
model.specification
# Specifications(problem=<Problem.MULTI_LABEL_CLASSIFICATION: 2>, 
#                resolution=<Resolution.FRAME: 1>, 
#                duration=6, 
#                warm_up=(0.0, 0.0), 
#                classes=['speech'], 
#                permutation_invariant=False)

Would be nice to update classes to ['vad', 'snr', 'c50'].

I fixed it with commit 192c0dd
I changed it in the pretrained model and in the code to avoid mismatch with future trainings.

Thanks for raising this issue!

Closing :)