
Episode 549: William Falcon Optimizing Deep Learning Models
Software Engineering Radio - the podcast for professional software developers
00:00
Using Lightning to Train Machine Learning Models
The traditional PyTorch lightning module trainer integrates with any experiment manager that's out there. So you can find two models from any library you want, open AI and so on. It's like three lines of code to find 200 train these things, torch vision, et cetera. And then as I mentioned, now we've evolved the project that's called lightning now. With the lightning work and the lightning flow, you can actually integrate any framework, even like TensorFlow. We have all the accelerators, so GPUs, HPU, TPUs. You can do jacks. As long as it's machine learning, you can do numpy, sklearn, all
Transcript
Play full episode