d04c9e9ba382be370eba671b6e5efc3403e24d6c,research/object_detection/builders/dataset_builder.py,,read_dataset,#,109

Before Change

    tf.logging.info("Sampling from datasets %s with weights %s" %
                    (input_files, config.sample_from_datasets_weights))
    records_datasets = []
    for input_file in input_files:
      records_dataset = _read_dataset_internal(file_read_func, [input_file],
                                               config, filename_shard_fn)
      records_datasets.append(records_dataset)
    dataset_weights = list(config.sample_from_datasets_weights)
    return tf.data.experimental.sample_from_datasets(records_datasets,
                                                     dataset_weights)
  else:

After Change

      if num_readers == 0:
        tf.logging.info("Skipping dataset due to zero weights: %s", input_file)
        continue
      tf.logging.info(
          "Num readers for dataset [%s]: %d", input_file, num_readers)
      records_dataset = _read_dataset_internal(file_read_func, [input_file],
                                               num_readers, config,
                                               filename_shard_fn)
      dataset_weights.append(weight)

Instances