mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 03:30:19 +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
4 lines
129 B
Python
4 lines
129 B
Python
from .fc_middleware import FCMiddleware
|
|
from .lstm_middleware import LSTMMiddleware
|
|
|
|
__all__ = ["FCMiddleware", "LSTMMiddleware"] |