From 087d4515c69c83d86dc8b8e7d135705ea734c09c Mon Sep 17 00:00:00 2001 From: Ethan White Date: Sat, 11 May 2024 11:45:44 -0400 Subject: [PATCH] Remove 'no validation file' warning This warning is triggered anytime a model is created and causes confusion for zero-shot users #664. Closes #664 --- deepforest/main.py | 1 - 1 file changed, 1 deletion(-) diff --git a/deepforest/main.py b/deepforest/main.py index 2e2249cc..1a635e19 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