- 保存和加载模型、优化器、随机数生成器和 GradScaler
- 使用
save_state()
将上述所有内容保存到一个文件夹位置 - 使用
load_state()
加载之前通过save_state()
保存的状态
- 使用
- 通过使用
register_for_checkpointing()
,可以注册自定义对象以便自动从前两个函数中存储或加载- 只要对象具有
state_dict
和load_state_dict
功能即可 - 这可以包括诸如学习率调度器之类的对象。
- 只要对象具有
save_state()
将上述所有内容保存到一个文件夹位置load_state()
加载之前通过 save_state()
保存的状态register_for_checkpointing()
,可以注册自定义对象以便自动从前两个函数中存储或加载
state_dict
和 load_state_dict
功能即可本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:/a/676461.html
如若内容造成侵权/违法违规/事实不符,请联系我们进行投诉反馈qq邮箱809451989@qq.com,一经查实,立即删除!