mirror of
https://github.com/vale981/ray
synced 2025-03-10 05:16:49 -04:00
3e6ba5d6d2
* Revert "Revert "[RLlib] POC: `PGTrainer` class that works by sub-classing, not `trainer_template.py`. (#20055)" (#20284)"
This reverts commit 246787cdd9.
Co-authored-by: sven1977 <svenmika1977@gmail.com>
16 lines
398 B
Python
16 lines
398 B
Python
from ray.rllib.agents.trainer import with_common_config
|
|
|
|
# yapf: disable
|
|
# __sphinx_doc_begin__
|
|
|
|
# Add the following (PG-specific) updates to the (base) `Trainer` config in
|
|
# rllib/agents/trainer.py (`COMMON_CONFIG` dict).
|
|
DEFAULT_CONFIG = with_common_config({
|
|
# No remote workers by default.
|
|
"num_workers": 0,
|
|
# Learning rate.
|
|
"lr": 0.0004,
|
|
})
|
|
|
|
# __sphinx_doc_end__
|
|
# yapf: enable
|