search

slSeanWU / beats-conformer-bart-audio-captioner

PyTorch implementation of the ICASSP-24 paper: "Improving Audio Captioning Models with Fine-grained Audio Features, Text Embedding Supervision, and LLM Mix-up Augmentation"
Apache License 2.0
29 stars 1 forks source link

Is this code incomplete？ #3

Open yangdongdong2000 opened 2 months ago

yangdongdong2000 commented 2 months ago

There is no code related with training, is this code incomplete or there are no plans to open source the training code

slSeanWU commented 2 months ago

Hi @yangdongdong2000 ,

Thanks for reaching out. Unfortunately, we don't plan to release the training script.

We have the data loaders ready under data/, with which you could easily plug in custom data. We recommend using HuggingFace trainer if you'd like to train your own models. Thanks again!