Introduce TD3

ShangtongZhang · Aug 2, 2019 · 4c2168b · 4c2168b
1 parent 00b957f
commit 4c2168b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/deep_rl/agent/DDPG_agent.py b/deep_rl/agent/DDPG_agent.py
@@ -10,7 +10,7 @@
 import torchvision
 
 
-class DDPGAgent(BaseAgent):
+class TD3(BaseAgent):
     def __init__(self, config):
         BaseAgent.__init__(self, config)
         self.config = config

diff --git a/deep_rl/agent/__init__.py b/deep_rl/agent/__init__.py
@@ -6,3 +6,4 @@
 from .QuantileRegressionDQN_agent import *
 from .PPO_agent import *
 from .OptionCritic_agent import *
+from .TD3_agent import *