refactor dir structure

This commit is contained in:
2024-06-02 11:13:04 +02:00
parent 8a013a4746
commit b34224f189
4 changed files with 3 additions and 2 deletions
+1 -1
View File
@@ -4,6 +4,6 @@ try:
except ImportError: except ImportError:
pass pass
from fancy_rl.ppo import PPO from fancy_rl.algos import PPO
__all__ = ["PPO"] __all__ = ["PPO"]
+1
View File
@@ -0,0 +1 @@
from fancy_rl.algos.ppo import PPO
+1 -1
View File
@@ -2,7 +2,7 @@ import torch
from torchrl.modules import ActorValueOperator, ProbabilisticActor from torchrl.modules import ActorValueOperator, ProbabilisticActor
from torchrl.objectives import ClipPPOLoss from torchrl.objectives import ClipPPOLoss
from torchrl.objectives.value.advantages import GAE from torchrl.objectives.value.advantages import GAE
from fancy_rl.on_policy import OnPolicy from fancy_rl.algos.on_policy import OnPolicy
from fancy_rl.policy import Actor, Critic, SharedModule from fancy_rl.policy import Actor, Critic, SharedModule
class PPO(OnPolicy): class PPO(OnPolicy):