hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-05 10:01:43 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[RLlib] Fix config mismatch for train_one_step. num_sgd_iter instead of sgd_num_iter. (#21555)

Browse source
This commit is contained in:
Vince Jankovics 2022-01-18 15:00:27 +00:00 • committed by GitHub
parent fa5c167717
commit 7dc3de4eed
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
rllib/execution/train_ops.py
View file

@ -34,7 +34,7 @@ def train_one_step(trainer, train_batch) -> Dict:
workers = trainer.workers
local_worker = workers.local_worker()
policies = local_worker.policies_to_train
num_sgd_iter = config.get("sgd_num_iter", 1)
num_sgd_iter = config.get("num_sgd_iter", 1)
sgd_minibatch_size = config.get("sgd_minibatch_size", 0)
learn_timer = trainer._timers[LEARN_ON_BATCH_TIMER]