hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-14 07:06:38 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions
ray/release/air_tests/air_benchmarks/workloads
History
Kai Fricke 213a96e239
[air/benchmarks] Add distributed Tensorflow benchmarks (CPU only) (#26519) 
Following up from #26436, this PR adds a distributed benchmark test for Tensorflow FashionMNIST training. It compares training with Ray AIR with training with vanilla PyTorch.

Signed-off-by: Kai Fricke <kai@anyscale.com>
2022-07-14 22:08:43 +01:00
..
_tensorflow_prepare.py [air/benchmarks] Add distributed Tensorflow benchmarks (CPU only) (#26519) 2022-07-14 22:08:43 +01:00
_torch_prepare.py [air] Add AIR distributed training benchmark for Torch FashionMNIST (#26436) 2022-07-13 10:53:24 +01:00
benchmark_util.py [air/benchmarks] Add distributed Tensorflow benchmarks (CPU only) (#26519) 2022-07-14 22:08:43 +01:00
tensorflow_benchmark.py [air/benchmarks] Add distributed Tensorflow benchmarks (CPU only) (#26519) 2022-07-14 22:08:43 +01:00
torch_benchmark.py [air/benchmarks] Add distributed Tensorflow benchmarks (CPU only) (#26519) 2022-07-14 22:08:43 +01:00