mirror of
https://github.com/gryf/coach.git
synced 2026-01-17 04:14:21 +01:00
5fadb9c18e
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
15 lines
247 B
Python
15 lines
247 B
Python
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'
|
|
]
|