added direct links to docs

2026-09-10 12:21:57 +08:00 · 2020-01-16 07:27:31 -05:00
parent 637f2344de
commit 3e5c5f9588
1 changed files with 4 additions and 0 deletions
@@ -18,3 +18,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']