hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[air] add pytorch benchmark number (#26719)

Browse source
This commit is contained in:
Richard Liaw 2022-07-19 09:51:13 -07:00 • committed by GitHub
parent 03eb9f153c
commit 6563c2762d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
doc/source/ray-air/benchmarks.rst
View file

@ -193,7 +193,7 @@ We demonstrate that the performance is similar between the two frameworks.
- `python workloads/torch_benchmark.py run --num-runs 3 --num-epochs 20 --num-workers 4 --cpus-per-worker 8`
* - 4 g4dn.12xlarge node (16 workers)
- FashionMNIST
- TODO
- 236.61 s (vs 220.97 s Pytorch)
- `python workloads/torch_benchmark.py run --num-runs 3 --num-epochs 20 --num-workers 16 --cpus-per-worker 4 --use-gpu`