search

JusperLee / CTCNet

An Audio-Visual Speech Separation Model Inspired by Cortico-Thalamo-Cortical Circuits
Apache License 2.0
69 stars 16 forks source link

Error for pytorch-lightning #10

Closed wcycqjy closed 8 months ago

wcycqjy commented 8 months ago

Thanks for sharing your excellent work! But I confronted an error and thus asking for help. When I run the trainer.fit(system) in train_ctc.py, there is an error said: The LightningModule.on_epoch_end hook was removed in v1.8. Please use LightningModule.on_<train/validation/test>_epoch_end instead.

But I can't find any possible code in train_ctc.py to fix it. I think this happens to the wrong version of pytorch-lightning, but the version is exactly what you wrote in the readme. So can you help me fix this problem?

JusperLee commented 8 months ago

8

wcycqjy commented 8 months ago

Thanks for your reply. I see that issue but get confused where to put the new code def on_validation_epoch_end(self). Could you explain it more explicitly?

wcycqjy commented 8 months ago

I find it, thanks.