From 765d8fc5b268a948e19e38fb0c8884d1b0b051e1 Mon Sep 17 00:00:00 2001 From: Antonin Raffin Date: Thu, 12 Mar 2020 13:24:11 +0100 Subject: [PATCH] Fix event callback --- torchy_baselines/common/callbacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: