6f36525c5da0fe912c89790a51b9ea9e7cc3d110,tensorflow_datasets/scripts/benchmark_dataset.py,,main,#Any#,40

Before Change

  ds_name = FLAGS.dataset
  data_dir = FLAGS.data_dir
  epochs = FLAGS.epochs
  benchmark(ds_name, data_dir, epochs)


if __name__ == "__main__":
  app.run(main)

After Change

  ds = _preprocess(ds_name, data_dir)

  for split in ds:
    print("Benchmarking split:", split)
    print("\n")
    benchmark(ds[split], epochs)


if __name__ == "__main__":
  app.run(main)

Instances