提供的ckpt是比赛中的吗？

wsntxxn / AudioCaption

Audio captioning recipe

MIT License

43 stars 4 forks source link

Closed faychu closed 4 months ago

faychu commented 1 year ago

您好，想问下这个repo中提供的ckpt就是比赛中提交的吗？

wsntxxn commented 1 year ago

比赛提交的是多个模型的ensemble，每个都比较大，为了方便，这里就放了一个用CNN14-BERT medium的retrieval model作为feature extractor，后面接3层RNN encoder和2层Transformer decoder的ckpt