Internal change

PiperOrigin-RevId: 278756245

official/benchmark/retinanet_benchmark.py

@@ -134,7 +134,10 @@ class RetinanetBenchmarkBase(DetectionBenchmarkBase):
 
   def _run_detection_main(self):
     """Starts detection job."""
+    if self.timer_callback:
       return detection.run(callbacks=[self.timer_callback])
+    else:
+      return detection.run()
 
 
 class RetinanetAccuracy(RetinanetBenchmarkBase):
@@ -180,6 +183,10 @@ class RetinanetAccuracy(RetinanetBenchmarkBase):
             'iterations_per_loop': 100,
             'total_steps': 22500,
             'train_file_pattern': self.train_data_path,
+            'checkpoint': {
+                'path': self.resnet_checkpoint_path,
+                'prefix': 'resnet50/'
+            },
         },
         'eval': {
             'batch_size': 8,

official/modeling/training/distributed_executor.py

@@ -426,6 +426,7 @@ class DistributedExecutor(object):
       if not custom_callbacks:
         return
       for callback in custom_callbacks:
+        if callback:
           callback.on_batch_begin(batch)
 
     def _run_callbacks_on_batch_end(batch):
@@ -433,6 +434,7 @@ class DistributedExecutor(object):
       if not custom_callbacks:
         return
       for callback in custom_callbacks:
+        if callback:
           callback.on_batch_end(batch)
 
     if save_config: