模型文件大

[baisong666]

第一步训练生成的4个模型文件的大小为256M，第二步生成的模型文件大小为500M左右，请问为啥会折磨大呢，这不是要部署在手机上吗，这样大的模型文件怎麼部署呢，请作者看到后能不能回复一下，万分感谢

[baisong666]

你好，可以看看我的问题吗

[baisong666]

可以看看不

[baisong666]

救命阿

[baisong666]

help

[nightsnack]

看到了。第二步的模型文件里面存了还有optimizer log之类的东西所以大了。

YOLObile/train.py

Line 667 in 2494a48

torch.save(chkpt, last)

，你改一下。这里只存model。部署手机需要compiler，compiler没有开源

[baisong666]

是改成torch.save(last) 和 torch.save(best)吗

[baisong666]

我设置了name后，模型变成了256M，现在按照你的方法ckpt那个只有model了，模型还是256M，这又是为啥呢😂

…

---原始邮件--- 发件人: ***@***.***> 发送时间: 2021年7月13日(周二) 下午4:10 收件人: ***@***.***>; 抄送: ***@***.******@***.***>; 主题: Re: [nightsnack/YOLObile] 模型文件大 (#15) 看到了。第二步的模型文件里面存了还有optimizer log之类的东西所以大了。https://github.com/nightsnack/YOLObile/blob/2494a48ff3591da3b400dcb3c531a77d22d510e4/train.py#L667，你改一下。这里只存model。部署手机需要compiler，compiler没有开源 — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[baisong666]

chkpt = {#'epoch': epoch,
#'best_fitness': best_fitness,
#'training_results': f.read(),
'model': ema.ema.module.state_dict() if hasattr(model,
'module') else ema.ema.state_dict()
#'optimizer': None if final_epoch else optimizer.state_dict()
}

[baisong666]

是这样把其他的注释掉吗

[baisong666]

预训练模型选择哪一个

[baisong666]

help

[nightsnack]

256M有什么问题吗？这是存的.pt，pruning之后sparse的model里面0也是存下来的呀。在手机上运行的时候才会重新按照index-val的方式重新存储

[baisong666]

啥意思哈哈，是咱这个程序训练出来就是256M吗

[baisong666]

可不可以具体说一说呢