wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-29 11:26:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove type information from remote decorator.

Browse Source
This commit is contained in:
Robert Nishihara
2016-08-29 22:05:59 -07:00
parent 93e6c9947b
commit d7f313a026
23 changed files with 147 additions and 305 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/hyperopt/hyperopt.py
+1 -1
View File
@@ -51,7 +51,7 @@ def cnn_setup(x, y, keep_prob, lr, stddev):
# Define a remote function that takes a set of hyperparameters as well as the
# data, consructs and trains a network, and returns the validation accuracy.
@ray.remote([dict, int, np.ndarray, np.ndarray, np.ndarray, np.ndarray], [float])
@ray.remote()
def train_cnn_and_compute_accuracy(params, steps, train_images, train_labels, validation_images, validation_labels):
# Extract the hyperparameters from the params dictionary.
learning_rate = params["learning_rate"]