Skip to content

Conversation

@LennartPurucker
Copy link
Contributor

@LennartPurucker LennartPurucker commented Nov 23, 2024

  • Renames pytorch lightning to the current name
  • adds timelimit
  • removes numpy msgpack which breaks ray (!)
  • make skorch a lazy import

Rel. the lightning changes: Lightning-AI/pytorch-lightning#16688
(I am not sure if it is a good idea to switch, but I had crashes without this change)

This is just a PR to show some suggestions, but it is likely not ideal for your code / or well-implemented. Mainly to start a documented discussion.

@dholzmueller dholzmueller merged commit 0e7b41c into dholzmueller:main Nov 23, 2024
12 checks passed
@dholzmueller
Copy link
Owner

Thank you for the effort! I changed the imports to fall back to pytorch_lightning if lightning isn't available, and improved some minor things.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants