Lets say i got a pretrained model model.pt

how do i know what classes can this model predict?

i think it is saved inside the model but how do I extract it?

Im trying to understand what https://github.com/AndreyGuzhov/AudioCLIP does

it has a pretrained AudioCLIP-Full-Training.pt

how do I know the labels or classes inside this AudioCLIP-Full-Training.pt

CodePudding user response：

take a look at its demo notebook: https://github.com/AndreyGuzhov/AudioCLIP/blob/master/demo/AudioCLIP.ipynb.

As a CLIP model, the labels used in classification is an model inputs.

CodePudding user response：

As @lauthu already said, the first place to look would be the Notebook: https://github.com/AndreyGuzhov/AudioCLIP/blob/master/demo/AudioCLIP.ipynb.

The notebook mentions these labels

LABELS = ['cat', 'thunderstorm', 'coughing', 'alarm clock', 'car horn']