change the order of args

weecology · Nov 9, 2023 · 6050220 · 6050220
1 parent fd93aab
commit 6050220
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepforest/callbacks.py b/deepforest/callbacks.py
@@ -63,7 +63,7 @@ def log_images(self, pl_module):
                   "skipping upload, images were saved to {}, "
                   "error was rasied {}".format(self.savedir, e))
 
-    def on_validation_epoch_end(self, trainer, pl_module):
+    def on_validation_end(self, trainer, pl_module):
         if trainer.sanity_checking:  # optional skip
             return