Replies: 2 comments
-
|
Hey @wjgancn, thanks for the kind words 😊 At the moment, we are not planning to provide further support for torch lightning to avoid as much as possible additional dependencies (even optional). However, if you are keen using torch lightning, I think the current modules implementation should be compatible. Are you currently witnessing anything preventing you to use torch lightning with deepinv? |
Beta Was this translation helpful? Give feedback.
0 replies
-
|
Would Lightning Fabric be a good compromise? |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hello,
Thank you for creating such an amazing repository. I am just curious if there are any plans to make this repository compatible with the PyTorch Lightning Trainer (https://lightning.ai/docs/pytorch/stable/api_references.html#trainer). The Lightning Trainer functions similarly to the existing trainer in this repository but offers many additional options, particularly useful for distributed training. I think a straightforward way to achieve compatibility could be instancing the model of torch.nn.Module in the LightningModule class.
Thanks!
Beta Was this translation helpful? Give feedback.
All reactions