From 3caa3805f04e9aacc260b99984bba46117ce6996 Mon Sep 17 00:00:00 2001 From: Dominik Jain Date: Thu, 14 Dec 2023 11:47:32 +0100 Subject: [PATCH] Fix: SamplingConfig.start_timesteps_random was not used --- tianshou/highlevel/agent.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tianshou/highlevel/agent.py b/tianshou/highlevel/agent.py index dce3379..772c9d1 100644 --- a/tianshou/highlevel/agent.py +++ b/tianshou/highlevel/agent.py @@ -115,7 +115,10 @@ class AgentFactory(ABC, ToStringMixin): train_collector = Collector(policy, train_envs, buffer, exploration_noise=True) test_collector = Collector(policy, envs.test_envs) if self.sampling_config.start_timesteps > 0: - train_collector.collect(n_step=self.sampling_config.start_timesteps, random=True) + train_collector.collect( + n_step=self.sampling_config.start_timesteps, + random=self.sampling_config.start_timesteps_random, + ) return train_collector, test_collector def set_policy_wrapper_factory(