FastPitchModel_SSL/SSLDisentangler 在验证时崩溃: on_validation_epoch_end() 缺少 1 个必需的位置参数: 'outputs'
作者: udsy19创建于 2026年9月5日更新于 2026年9月16日
标签community-request
描述错误
FastPitchModel_SSL 和 SSLDisentangler(nemo/collections/tts/models/fastpitch_ssl.py,
nemo/collections/tts/models/ssl_tts.py) 定义了 on_validation_epoch_end(self, outputs) 函数,这是 Lightning 2.0 之前的钩子签名。Lightning 的评估循环调用此钩子,带有 零 个位置参数(_EvaluationLoop._on_evaluation_epoch_end ->
call._call_lightning_module_hook(trainer, "on_validation_epoch_end")),因此无论哪个模型都在真正的验证 epoch 中引发 TypeError 之前,用户代码都不会运行,从而在第一次合理的验证迭代中中止训练。
内容来源: NVIDIA-NeMo/Speech