dodox/fancy_rl
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

refactor dir structure

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-06-02 11:13:04 +02:00
parent 8a013a4746
commit b34224f189
4 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fancy_rl/__init__.py
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
fancy_rl/algos/__init__.py Normal file
View File

@ -0,0 +1 @@
from fancy_rl.algos.ppo import PPO

0
fancy_rl/on_policy.py → fancy_rl/algos/on_policy.py
View File

2
fancy_rl/ppo.py → fancy_rl/algos/ppo.py
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):