mirror of
https://github.com/wassname/ray.git
synced 2026-09-09 11:32:43 +08:00
Test example applications and rllib in jenkins tests. (#707)
* Test example applications in Jenkins. * Fix default upload_dir argument for Algorithm class. * Fix evolution strategies. * Comment out policy gradient example which doesn't seem to work. * Set --env-name for evolution strategies.
This commit is contained in:
committed by
Philipp Moritz
parent
4349f1f966
commit
80e8426b5e
@@ -150,7 +150,9 @@ if __name__ == "__main__":
|
||||
help="The number of rollouts to do per batch.")
|
||||
parser.add_argument("--redis-address", default=None, type=str,
|
||||
help="The Redis address of the cluster.")
|
||||
|
||||
parser.add_argument("--iterations", default=-1, type=int,
|
||||
help="The number of model updates to perform. By "
|
||||
"default, training will not terminate.")
|
||||
args = parser.parse_args()
|
||||
batch_size = args.batch_size
|
||||
|
||||
@@ -169,7 +171,9 @@ if __name__ == "__main__":
|
||||
# Update the rmsprop memory.
|
||||
rmsprop_cache = {k: np.zeros_like(v) for k, v in model.items()}
|
||||
actors = [PongEnv.remote() for _ in range(batch_size)]
|
||||
while True:
|
||||
iteration = 0
|
||||
while iteration != args.iterations:
|
||||
iteration += 1
|
||||
model_id = ray.put(model)
|
||||
actions = []
|
||||
# Launch tasks to compute gradients from multiple rollouts in parallel.
|
||||
|
||||
Reference in New Issue
Block a user