Callbacks [wip] (#889)

* Add callback system + associated test * Add trainer and pl_module args to callback methods * typing * typo in docstring * Switch to on_.*_start() * fix on_test_start * fix the mess after rebasing
2026-09-09 11:32:07 +08:00 · 2020-02-25 23:17:27 -05:00
parent 96b058c5fa
commit be244560b2
14 changed files with 407 additions and 87 deletions
@@ -9,6 +9,6 @@ Callbacks
        _save_model,
        on_epoch_end,
        on_train_end,
-        on_epoch_begin,
+        on_epoch_start,
        check_monitor_top_k,
-        on_train_begin,
+        on_train_start,
@@ -9,4 +9,4 @@ Loggers
        _save_model,
        on_epoch_end,
        on_train_end,
-        on_epoch_begin,
+        on_epoch_start,