NotImplementedError: Support for `validation_epoch_end` has been removed in v2.0.0. `ReconstructionTrainer` implements this method. You can use the `on_validation_epoch_end` hook instead. To access outputs, save them in-memory as instance attributes. You can find migration examples in https://github.com/Lightning-AI/lightning/pull/16520.
时间: 2023-06-16 15:05:39 浏览: 1116
这个错误是因为你正在使用 PyTorch Lightning v2.0.0 或更高版本,并且你的代码中使用了 `validation_epoch_end` 方法,但是在这个版本中已经不支持这个方法了。相反,你可以使用 `on_validation_epoch_end` 钩子来替代它。 钩子函数中可以将输出结果保存为实例属性,然后在需要的时候访问。你可以参考 https://github.com/Lightning-AI/lightning/pull/16520 中的迁移示例来修改你的代码。
阅读全文