Pytorch 闪电模型有两个不同的钩子:
从文档:
setup
- 在适合和测试开始时调用。on_fit_start
- 在合身开始时调用。如果在 DDP 上,它会在每个进程上调用。
我无法理解这些差异,以及何时使用其中任何一个。
我还注意到一些类变量仍然没有定义setup
在self.datamodule
.
setup
vs的预期用例是on_fit_start
什么?
Pytorch 闪电模型有两个不同的钩子:
从文档:
setup
- 在适合和测试开始时调用。on_fit_start
- 在合身开始时调用。如果在 DDP 上,它会在每个进程上调用。我无法理解这些差异,以及何时使用其中任何一个。
我还注意到一些类变量仍然没有定义setup
在self.datamodule
.
setup
vs的预期用例是on_fit_start
什么?