Closed mirkobronzi closed 4 years ago
Add support for pytorch lightning and keras. (so, in short, replacing the train loop with these more high-level API)
Add support for pytorch lightning and keras. (so, in short, replacing the train loop with these more high-level API)