зеркало из https://github.com/microsoft/qlib.git
Add exploration noise to rl training collector (#1481)
* Update vessel.py Add exploration_noise=True to training collector * Update vessel.py Reformat
This commit is contained in:
Родитель
05d67b3828
Коммит
5387ea5c1f
|
@ -168,7 +168,9 @@ class TrainingVessel(TrainingVesselBase):
|
|||
self.policy.train()
|
||||
|
||||
with vector_env.collector_guard():
|
||||
collector = Collector(self.policy, vector_env, VectorReplayBuffer(self.buffer_size, len(vector_env)))
|
||||
collector = Collector(
|
||||
self.policy, vector_env, VectorReplayBuffer(self.buffer_size, len(vector_env)), exploration_noise=True
|
||||
)
|
||||
|
||||
# Number of episodes collected in each training iteration can be overridden by fast dev run.
|
||||
if self.trainer.fast_dev_run is not None:
|
||||
|
|
Загрузка…
Ссылка в новой задаче