Jihuai commited on
Commit
baa7a43
·
1 Parent(s): 9ca5133

update readme train todo

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -19,7 +19,7 @@ license: mit
19
 
20
  - Some utility functions for inference.
21
 
22
- - (TODO) Train on more datasets.
23
 
24
  ## Model weights
25
  You need to download the model weights from [huggingface model](https://huggingface.co/chenxie95/Language-Audio-Banquet-ckpt) and put them in `checkpoints/`. `bandit-vdbo-roformer.ckpt` is needed for training. `ev-pre.ckpt` and `ev-pre-aug.ckpt` can be choosen for inference.
 
19
 
20
  - Some utility functions for inference.
21
 
22
+ - (TODO) Train on more datasets; language-audio joint training; fine-tuning for CLAP.
23
 
24
  ## Model weights
25
  You need to download the model weights from [huggingface model](https://huggingface.co/chenxie95/Language-Audio-Banquet-ckpt) and put them in `checkpoints/`. `bandit-vdbo-roformer.ckpt` is needed for training. `ev-pre.ckpt` and `ev-pre-aug.ckpt` can be choosen for inference.