Skip to content

Commit 5387ea5

Browse files
authored
Add exploration noise to rl training collector (#1481)
* Update vessel.py Add exploration_noise=True to training collector * Update vessel.py Reformat
1 parent 05d67b3 commit 5387ea5

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

qlib/rl/trainer/vessel.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,9 @@ def train(self, vector_env: FiniteVectorEnv) -> Dict[str, Any]:
168168
self.policy.train()
169169

170170
with vector_env.collector_guard():
171-
collector = Collector(self.policy, vector_env, VectorReplayBuffer(self.buffer_size, len(vector_env)))
171+
collector = Collector(
172+
self.policy, vector_env, VectorReplayBuffer(self.buffer_size, len(vector_env)), exploration_noise=True
173+
)
172174

173175
# Number of episodes collected in each training iteration can be overridden by fast dev run.
174176
if self.trainer.fast_dev_run is not None:

0 commit comments

Comments
 (0)