wassname/ray
1
0 Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-10-03 12:50:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] Improve mnist_pytorch.py example (#3894)

Browse Source 
## What do these changes do?

* Improved --no-cuda handling
* Removed deprecated Variable usage


## Related issue number

Fixes #3873 
<!-- Are there any issues opened that will be resolved by merging this change? -->
This commit is contained in:
vfdev authored and Richard Liaw committed 2019-02-04 17:59:54 -08:00
1 parent 5fb813ff39
commit b2b8417790
3 files changed
+28 -31
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

test/jenkins_tests/run_multi_node_tests.sh
+2 -3
View File
@@ -367,12 +367,11 @@ docker run --rm --shm-size=${SHM_SIZE} --memory=${MEMORY_SIZE} $DOCKER_SHA \
--smoke-test
docker run --rm --shm-size=${SHM_SIZE} --memory=${MEMORY_SIZE} $DOCKER_SHA \
python /ray/python/ray/tune/examples/mnist_pytorch.py \
--smoke-test
python /ray/python/ray/tune/examples/mnist_pytorch.py --smoke-test --no-cuda
docker run --rm --shm-size=${SHM_SIZE} --memory=${MEMORY_SIZE} $DOCKER_SHA \
python /ray/python/ray/tune/examples/mnist_pytorch_trainable.py \
--smoke-test
--smoke-test --no-cuda
docker run --rm --shm-size=${SHM_SIZE} --memory=${MEMORY_SIZE} $DOCKER_SHA \
python /ray/python/ray/tune/examples/genetic_example.py \