Extracting embeddings.

[audio-westlakeu]

Extracting embeddings.

[lmaxwell]

Hi，@Sreeni1204. Thanks for you insterest in our work!

For extracting embedding, please see example in "/audiossl/methods/atst/downstream/train_freeze.py".

pretrained_encoder = get_pretraied_encoder(args)

pretrained_module = PretrainedEncoderPLModule(pretrained_encoder,
                                                    6.,
                                                    args.n_last_blocks)
embedding = pretrained_module(input)

You may also want see extract_embedding function(also in train_freeze.py), which show how to extract all embeddings from a dataloader.

Hope this helps. If you have any further questions, please let me know.

[Sreeni1204]

Hello @lmaxwell thank you for the clarifications. I will go through the file, and consult you for further clarifications.

Thank you :)