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:
Di 2023-08-18 17:41:02 +08:00 коммит произвёл GitHub
Родитель 05d67b3828
Коммит 5387ea5c1f
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 3 добавлений и 1 удалений

Просмотреть файл

@ -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: