0

Pytorch 闪电模型有两个不同的钩子

从文档:

  1. setup- 在适合和测试开始时调用。
  2. on_fit_start- 在合身开始时调用。如果在 DDP 上,它会在每个进程上调用。

我无法理解这些差异,以及何时使用其中任何一个。

我还注意到一些类变量仍然没有定义setupself.datamodule.


setupvs的预期用例是on_fit_start什么?

4

0 回答 0