mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 11:40:18 +01: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:
@@ -5,12 +5,12 @@ COPY setup.py /root/src/.
|
||||
COPY requirements.txt /root/src/.
|
||||
COPY README.md /root/src/.
|
||||
WORKDIR /root/src
|
||||
RUN pip3 install -e .
|
||||
RUN pip3 install -e .[all]
|
||||
|
||||
# everything above here should be cached most of the time
|
||||
COPY . /root/src
|
||||
WORKDIR /root/src
|
||||
RUN pip3 install -e .
|
||||
RUN pip3 install -e .[all]
|
||||
|
||||
RUN chmod 777 /root/src/docker/docker_entrypoint.sh
|
||||
ENTRYPOINT ["/root/src/docker/docker_entrypoint.sh"]
|
||||
|
||||
Reference in New Issue
Block a user