Update README.md

add repo & training datasets

README.md

@@ -17,7 +17,13 @@ This repo contains the config & pretrained weights of the model described in the
   [[arXiv page](https://arxiv.org/abs/2309.17352)]
 ## GitHub Repository
 To use this model, please refer to our code published at:
-- TBA
+- https://github.com/slSeanWU/beats-conformer-bart-audio-captioner
+## Training Data
+- Pretrain
+  - **AudioCaps**: https://github.com/cdjkim/audiocaps/tree/master
+  - **ChatGPT mix-ups from Clotho**: https://huggingface.co/datasets/slseanwu/clotho-chatgpt-mixup-50K
+- Finetune
+  - **Clotho (V2)**: https://zenodo.org/records/4783391 
 ## BibTex
 If you find our model useful, please consider citing our paper. Thanks!
 ```