Reduced the number of statements

openvinotoolkit · Aug 17, 2023 · 46e6449 · 46e6449
1 parent 45948b2
commit 46e6449
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/examples/tensorflow/object_detection/main.py b/examples/tensorflow/object_detection/main.py
@@ -323,8 +323,7 @@ def run(config):
 
     # Training parameters
     epochs = config.epochs
-    steps_per_epoch = train_builder.steps_per_epoch
-    num_test_batches = test_builder.steps_per_epoch
+    steps_per_epoch, num_test_batches = train_builder.steps_per_epoch, test_builder.steps_per_epoch
 
     # Create model builder
     model_builder = get_model_builder(config)
@@ -335,10 +334,9 @@ def run(config):
         nncf_config=config.nncf_config, data_loader=train_dataset, batch_size=train_builder.global_batch_size
     )
 
-    resume_training = config.ckpt_path is not None
-
     compression_state = None
-    if resume_training:
+    if config.ckpt_path is not None:
+        # Resume training
         compression_state = load_compression_state(config.ckpt_path)
 
     with TFModelManager(model_builder.build_model, config.nncf_config, weights=config.get("weights", None)) as model: