61a619024a39e104d35911a541182de0eb182bde,official/vision/image_classification/dataset_factory.py,DatasetBuilder,build,#DatasetBuilder#Any#,267

Before Change

    builder = builders.get(self.config.builder, None)

    if builder is None:
      raise ValueError("Unknown builder type {}".format(self.config.builder))

    dataset = builder()
    dataset = self.pipeline(dataset, input_context)

    return dataset

After Change

      A TensorFlow dataset outputting batched images and labels.
    
    if strategy:
      if strategy.num_replicas_in_sync != self.config.num_devices:
        logging.warn("Passed a strategy with %d devices, but expected"
                     "%d devices.",
                     strategy.num_replicas_in_sync,
                     self.config.num_devices)

      dataset = strategy.experimental_distribute_datasets_from_function(
          self._build)
    else:
      dataset = self._build()

Instances