Closed huangjund closed 2 months ago
Thank you for your contribution! This code was submitted around two years ago, back when torch-lightning was still at version 1.x. I developed and tested it based on pytorch_lightning version 1.6.5. I have noticed that torch-lightning has now released version 2.x, but I have not fully tested my code on this new version and cannot guarantee that there will be no unexpected issues. Therefore, I will not be upgrading the dependencies at this time. However, you are very welcome to create a new branch to submit and update dependencies for others to reference.
solved the version problem. The code can now support cuda 11.8 and its related new version of pytorch and torch lightning.