mirror of
https://github.com/gryf/coach.git
synced 2026-04-10 15:13:40 +02:00
Adding mxnet components to rl_coach/architectures (#60)
Adding mxnet components to rl_coach architectures. - Supports PPO and DQN - Tested with CartPole_PPO and CarPole_DQN - Normalizing filters don't work right now (see #49) and are disabled in CartPole_PPO preset - Checkpointing is disabled for MXNet
This commit is contained in:
14
rl_coach/architectures/mxnet_components/heads/__init__.py
Normal file
14
rl_coach/architectures/mxnet_components/heads/__init__.py
Normal file
@@ -0,0 +1,14 @@
|
||||
from .head import Head, HeadLoss
|
||||
from .q_head import QHead
|
||||
from .ppo_head import PPOHead
|
||||
from .ppo_v_head import PPOVHead
|
||||
from .v_head import VHead
|
||||
|
||||
__all__ = [
|
||||
'Head',
|
||||
'HeadLoss',
|
||||
'QHead',
|
||||
'PPOHead',
|
||||
'PPOVHead',
|
||||
'VHead'
|
||||
]
|
||||
Reference in New Issue
Block a user