如何解决何时在 pytorch 闪电中使用 `on_fit_start` 与 `setup`?
Pytorch 闪电模型有两个不同的 hooks:
来自文档:
-
setup
- 在拟合和测试开始时调用。 -
on_fit_start
- 在 fit 开始时调用。如果在 DDP 上,它会在每个进程上调用。
我无法理解这些差异,以及何时使用。
我还注意到一些类变量仍未在 setup
上定义,例如 self.datamodule
。
setup
与 on_fit_start
的预期用例是什么?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。