Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions joyrl/algos/TD3/config.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
class AlgoConfig:
def __init__(self) -> None:
self.explore_steps = 1000 # exploration steps before training
self.explore_steps = 100 # exploration steps before training
self.policy_freq = 2 # policy update frequency
self.actor_lr = 1e-3 # actor learning rate 3e-4
self.actor_lr = 1e-4 # actor learning rate 3e-4
self.critic_lr = 1e-3 # critic learning rate
self.actor_hidden_dim = 256 # actor hidden layer dimension
self.critic_hidden_dim = 256 # critic hidden layer dimension
Expand All @@ -12,4 +12,4 @@ def __init__(self) -> None:
self.expl_noise = 0.1 # std of Gaussian exploration noise
self.noise_clip = 0.5 # range to clip target policy noise
self.batch_size = 100 # batch size for both actor and critic
self.buffer_size = 1000000 # replay buffer size
self.buffer_size = 1000000 # replay buffer size