Open qiyang77 opened 2 months ago
Thank you for reporting this issue. The compatibility between different versions of PyTorch Lightning (PL) can be troublesome, even between minor version changes. There is a straightforward solution that avoids such dependency issues. Notice that the Earthformer model and the pretrained weights are purely in PyTorch, without the dependencies on PL. You could implement your own training script with your specific version of PL. You may want to refer to the following scripts for guidance on how to use Earthformer purely in PyTorch: https://github.com/amazon-science/earth-forecasting-transformer/blob/7732b03bdb366110563516c3502315deab4c2026/tests/unittests/test_pretrained_checkpoints.py#L28
and
Taking the N-body minist script as an example, the code can't be executed with pytorch 2.0.1 due to the compatibility issue of pytorch_lightning.
Is there any workarounds to bypass these issues without installing an old version torch? Thx!
For pytorch_lightning=2.2.1
For pytorch_lightning=1.9.0
For pytorch_lightning=1.6.4