wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-09-11 12:43:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix Jenkins issue introduced by Variant Generator (#1194)

Browse Source 
* try fix

* shorten

* added a flag

* finish

* Fix linting.
This commit is contained in:
Richard Liaw
2017-11-09 00:56:20 -08:00
committed by Robert Nishihara
parent 1bf276cc08
commit 6197b260b8
2 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/examples/tune_mnist_ray.py
+9
View File
@@ -215,6 +215,15 @@ if __name__ == '__main__':
},
}
# These arguments are only for testing purposes.
parser = argparse.ArgumentParser()
parser.add_argument('--fast', action='store_true',
help='Run minimal iterations.')
args, _ = parser.parse_known_args()
if args.fast:
spec['stop']['training_iteration'] = 2
for trial in generate_trials(spec):
runner.add_trial(trial)
test/jenkins_tests/run_multi_node_tests.sh
+2 -4
View File
@@ -129,7 +129,5 @@ docker run --shm-size=10G --memory=10G $DOCKER_SHA \
python /ray/python/ray/rllib/test/test_checkpoint_restore.py
docker run --shm-size=10G --memory=10G $DOCKER_SHA \
python /ray/python/ray/tune/tune.py \
--env mnist \
--stop '{"training_iteration": 2}' \
--config '{"script_file_path": "/ray/python/ray/tune/examples/tune_mnist_ray.py", "activation": "relu"}'
python /ray/python/ray/tune/examples/tune_mnist_ray.py \
--fast