mirror of
https://github.com/wassname/ray.git
synced 2026-07-31 12:41:01 +08:00
* Working prototype * Pass buffer length, fix tests * Don't buffer per default * Dispatch and process save in one go, added tests * Fix tests * Pass adaptive seconds to train_buffered, stop result processing after STOP decision * Fix tests, add release test * Update tests * Added detailed logs for slow operations * Update python/ray/tune/trial_runner.py Co-authored-by: Richard Liaw <rliaw@berkeley.edu> * Apply suggestions from code review * Revert tests and go back to old tuning loop * nit Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
59 lines
1.0 KiB
Bash
Executable File
59 lines
1.0 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
ray_version=""
|
|
commit=""
|
|
ray_branch=""
|
|
|
|
for i in "$@"
|
|
do
|
|
echo "$i"
|
|
case "$i" in
|
|
--ray-version=*)
|
|
ray_version="${i#*=}"
|
|
|
|
;;
|
|
--commit=*)
|
|
commit="${i#*=}"
|
|
;;
|
|
--ray-branch=*)
|
|
ray_branch="${i#*=}"
|
|
;;
|
|
--workload=*)
|
|
workload="${i#*=}"
|
|
;;
|
|
--help)
|
|
usage
|
|
exit
|
|
;;
|
|
*)
|
|
echo "unknown arg, $i"
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
if [[ $ray_version == "" || $commit == "" || $ray_branch == "" ]]
|
|
then
|
|
echo "Provide --ray-version, --commit, and --ray-branch"
|
|
exit 1
|
|
fi
|
|
|
|
echo "version: $ray_version"
|
|
echo "commit: $commit"
|
|
echo "branch: $ray_branch"
|
|
echo "workload: ignored"
|
|
|
|
wheel="https://s3-us-west-2.amazonaws.com/ray-wheels/$ray_branch/$commit/ray-$ray_version-cp37-cp37m-manylinux2014_x86_64.whl"
|
|
|
|
pip install -U pip
|
|
pip install -U "$wheel"
|
|
pip install "ray[tune]" "ray"
|
|
pip install boto3==1.4.8 cython==0.29.0
|
|
|
|
if ! python "wait_cluster.py" 16 450; then
|
|
echo "Cluster did not come up in time. Aborting test."
|
|
exit 1
|
|
fi
|
|
|
|
python "workloads/$workload.py"
|