diff --git a/deepforest/main.py b/deepforest/main.py index d62cc6ce..ad1a2fba 100644 --- a/deepforest/main.py +++ b/deepforest/main.py @@ -197,7 +197,6 @@ def create_trainer(self, logger=None, callbacks=[], **kwargs): num_sanity_val_steps = 2 else: # Disable validation, don't use trainer defaults - print("No validation file provided. Turning off validation loop") limit_val_batches = 0 num_sanity_val_steps = 0 # Check for model checkpoint object