search

fschmid56 / EfficientAT

This repository aims at providing efficient CNNs for Audio Tagging. We provide AudioSet pre-trained models ready for downstream training and extraction of audio embeddings.
MIT License
218 stars 41 forks source link

More Compatible Teacher Pred #6

Closed didadida-r closed 1 year ago

didadida-r commented 1 year ago

Hi, i try to run the training script, and some data was loss during downloading. So the teacher pred index is not right and can not be used. I am wondering if your have time to update a Dict version of teacher pred with audio_name as key, because it would be more complex for new guy to get the ensemble logit

# here the teacher_preds is indexed by dataset index
if args.kd_lambda > 0:
      y_soft_teacher = teacher_preds[i]

Thanks

fschmid56 commented 1 year ago

Thanks for the request. This is already the third request regarding this. I'll put it on top of my list for next week to add the dict.

fschmid56 commented 1 year ago

I've uploaded the file fname_to_index, which contains a dict converting the file IDs to the indices in the predictions file. Tell me if this has the right format for you.

fschmid56 commented 1 year ago

As pointed out in this issue, there has been a problem with the _fname_toindex file. You can resolve this issue by re-downloading the file if you have already done so.