Inference

[AntonioArtur]

How do i get class predictions instead of embeddings?

[kkoutini]

Hi, I've added this feature in a new release 0.0.9, please update using:

pip install -e 'git+https://github.com/kkoutini/[email protected]#egg=hear21passt' 

After that you can get the logits (before the sigmoid activation) for the 527 classes, if you load the model like this:

model = load_model(mode="logits").cuda()

[faroit]

@kkoutini this is super useful. Maybe add this to the readme

[faroit]

@kkoutini is there a way to also load the ESC-50 pretrained models with passt_hear21?

[kkoutini]

@kkoutini is there a way to also load the ESC-50 pretrained models with passt_hear21?

@faroit I added more details for fine-tuning and loading the pre-trained models on ESC-50 here and the pre-trained models can be downloaded here.

[faroit]

@kkoutini Thanks!