diff --git a/torchy_baselines/common/callbacks.py b/torchy_baselines/common/callbacks.py index 1a8403a..b5a015d 100644 --- a/torchy_baselines/common/callbacks.py +++ b/torchy_baselines/common/callbacks.py @@ -129,7 +129,7 @@ class EventCallback(BaseCallback): def _on_event(self) -> bool: if self.callback is not None: - return self.callback() + return self.callback.on_step() return True def _on_step(self) -> bool: