added direct links to docs

2026-09-10 12:21:57 +08:00 · 2020-01-16 07:25:16 -05:00
parent 610edf8c3e
commit 9f9bf65ede
1 changed files with 4 additions and 0 deletions
@@ -22,3 +22,7 @@ The LightningTemplate holds the core computations, train, val, test loop, optimi

 This pattern de-couples the engineering from the science which makes your code reusable and free to run on any hardware.
 """
+
+from .trainer import Trainer
+
+__all__ = ['Trainer']