mirror of
https://github.com/wassname/ray.git
synced 2026-07-12 05:29:00 +08:00
Python 2 compatibility. (#5887)
This commit is contained in:
@@ -48,7 +48,7 @@ def random_one_hot_labels(shape):
|
||||
# Use GPU wth
|
||||
# @ray.remote(num_gpus=1)
|
||||
@ray.remote
|
||||
class Network():
|
||||
class Network(object):
|
||||
def __init__(self):
|
||||
self.model = create_keras_model()
|
||||
self.dataset = np.random.random((1000, 32))
|
||||
|
||||
Reference in New Issue
Block a user