Added a warning when starting training object detection model without…

… provided weights
openvinotoolkit · Aug 16, 2023 · 4ea3e9e · 4ea3e9e
1 parent ac6d9af
commit 4ea3e9e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/tensorflow/object_detection/main.py b/examples/tensorflow/object_detection/main.py
@@ -384,6 +384,8 @@ def run(config):
     test_step = create_test_step_fn(strategy, compress_model, predict_post_process_fn)
 
     if "train" in config.mode:
+        if config.weights is None:
+            logger.warning('Pretrained checkpoint is not provided. This may lead to poor training results!')
         if is_accuracy_aware_training(config):
             train_summary_writer = SummaryWriter(config.log_dir, "train")
             timer = Timer()