Chinese-CLIP 使用RN50预训练模型和flick30k后得到的权重文件特别大,

qcuzuvrc  于 2个月前  发布在  其他
关注(0)|答案(2)|浏览(43)

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

t8e9dugd

t8e9dugd1#

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

ctzwtxfj

ctzwtxfj2#

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

相关问题