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