search

OFA-Sys / Chinese-CLIP

Chinese version of CLIP which achieves Chinese cross-modal retrieval and representation generation.
MIT License
4k stars 418 forks source link

使用RN50预训练模型和flick30k后得到的权重文件特别大 #293

Open TC120 opened 3 months ago

TC120 commented 3 months ago

如题，官方提供的clip_cn_rn50.pt大小约300M，我训练完后得到的权重文件约900M，是因为保存时同时包含encode_image和encode_text模块吗？

fang520zhu commented 2 months ago

你好，请问训练后保存的权重文件具体位置在哪里

xxllp commented 2 months ago

保存的pt文件里面包含优化器的信息，所以变大了，可以去掉即可