GSTT-CSC / project-template

Template for new projects
0 stars 0 forks source link

Compatibility with Lightning Flash #20

Open hshuaib90 opened 3 years ago

hshuaib90 commented 3 years ago

Lightning Flash library looks to be really useful https://github.com/PyTorchLightning/lightning-flash for doing generic training tasks. It would be good if the Project_template was compatible with it.

laurencejackson commented 3 years ago

This looks cool, not heard of flash before. Seems like we would need there to be an appropriate network already defined somewhere to create a task from? Could be handy for testing common tasks like segmentation and classification. How do you see this being used?

I think there shouldn't need to be too many changes needed for compatibility since they say flash is just a stripped back but less flexible version of lightning with similar syntax. The main thing would probably be to try and use flash in the current template and see what changes we would need to make.