如题,官方提供的clip_cn_rn50.pt大小约为300M,我训练完后得到的权重文件约为900M,这是因为保存时同时包含了encode_image和encode_text模块吗?
t8e9dugd1#
你好,请问训练后保存的权重文件具体位置在哪里
ctzwtxfj2#
保存的pt文件里面包含优化器的信息,所以变大了,可以去掉即可。
2条答案
按热度按时间t8e9dugd1#
你好,请问训练后保存的权重文件具体位置在哪里
ctzwtxfj2#
保存的pt文件里面包含优化器的信息,所以变大了,可以去掉即可。