rename test_path -> worker_path (#128)

This commit is contained in:
Robert Nishihara
2016-06-19 22:01:31 -07:00
committed by Philipp Moritz
parent aaa0a514fa
commit ac65f1031e
7 changed files with 29 additions and 41 deletions
+2 -2
View File
@@ -15,8 +15,8 @@ parser.add_argument("--drop-ipython", default=False, type=bool, help="Drop into
if __name__ == "__main__":
args = parser.parse_args()
test_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "worker.py")
services.start_singlenode_cluster(return_drivers=False, num_workers_per_objstore=5, worker_path=test_path)
worker_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "worker.py")
services.start_singlenode_cluster(return_drivers=False, num_workers_per_objstore=5, worker_path=worker_path)
s3 = boto3.resource("s3")
imagenet_bucket = s3.Bucket(args.s3_bucket)
+2 -3
View File
@@ -19,9 +19,8 @@ num_batches = mnist.train.num_examples / batch_size
batches = [mnist.train.next_batch(batch_size) for _ in range(num_batches)]
if __name__ == "__main__":
test_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "worker.py")
test_path = os.path.join("worker.py")
services.start_singlenode_cluster(return_drivers=False, num_workers_per_objstore=16, worker_path=test_path)
worker_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "worker.py")
services.start_singlenode_cluster(return_drivers=False, num_workers_per_objstore=16, worker_path=worker_path)
x_batches = [ray.push(batches[i][0]) for i in range(num_batches)]
y_batches = [ray.push(batches[i][1]) for i in range(num_batches)]