KNN and Pretrained Models

[Sreyan88]

Hi there,

Thank you for the great repo! Can I please know if you guys have also open sourced the KNN code you have used after triplet loss trained representations and the pre-trained models on triplet loss?

Thank You

[roman-vygon]

Hello @Sreyan88!
Thank you for your interest to this repository.
The kNN implementation can be found in infer_train.py, specifically the FaissKNeighbors class. You can also find documentation on how to lower the memory footprint of the kNN model here.
Pre-trained models on google speech commands are in the /runs folder.

[Sreyan88]

Hi @roman-vygon ,

Thank you for your response! I am actually not being able to find the /runs folder you mentioned here. Could you please check?

[roman-vygon]

@Sreyan88 fixed it, sorry