Fix errors in classifier trainer when there is no distribution strategy

PiperOrigin-RevId: 306495321

Author: Zongwei Zhou

official/vision/image_classification/classifier_trainer.py

@@ -109,7 +109,8 @@ def _get_dataset_builders(params: base_configs.ExperimentConfig,
     logging.warning('label_smoothing not applied, so datasets will not be one '
                     'hot encoded.')
 
-  num_devices = strategy.num_replicas_in_sync
+  num_devices = strategy.num_replicas_in_sync if strategy else 1
+
   image_size = get_image_size_from_model(params)
 
   dataset_configs = [
@@ -308,7 +309,8 @@ def train_and_eval(
 
   strategy_scope = distribution_utils.get_strategy_scope(strategy)
 
-  logging.info('Detected %d devices.', strategy.num_replicas_in_sync)
+  logging.info('Detected %d devices.',
+               strategy.num_replicas_in_sync if strategy else 1)
 
   label_smoothing = params.model.loss.label_smoothing
   one_hot = label_smoothing and label_smoothing > 0