We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 63de8bf commit 498e981Copy full SHA for 498e981
examples/app_components/python/component_tracer.py
@@ -27,7 +27,7 @@ def on_train_start(self, trainer, pl_module) -> None:
27
print("Even the Lightning Work is available and state transfer works !")
28
print(self.lightning_work)
29
30
- def on_train_batch_end(self, trainer, *_) -> None:
+ def on_batch_end(self, trainer, *_) -> None:
31
# On every batch end, collects some information.
32
# This is communicated automatically to the rest of the app,
33
# so you can track your training in real time in the Lightning App UI.
0 commit comments