MLServer
MLServer copied to clipboard
Lightning Flash examples
Hey there,
Awesome framework. Would be nice to have a PyTorch Lightning or a Flash example (https://github.com/PyTorchLightning/lightning-flash).
Best, T.C
Hey @tchaton,
That would be really cool!
Would the loading of Lightning / Flash models be different to regular PyTorch models? Is there any particular feature of Lightning that you think would make sense to leverage in MLServer?
I think we're both based in the UK, so would be great to jump on a call to share some ideas and learn more about Lightning. I'll send you a separate email with some available times (I can just use the email listed in your GH profile - if that's alright).