26 lines (21 loc) · 699 Bytes

Deep reinforcement learning project 2

for course ME5406 deep learning for robotics

Table of contents

General info
Packages

General info

arm_gym.py: defines the environment
arm_unittests.py: unit tests for environment
NN.py: DDPG implementation
requirements.txt: required packages
Train_test.ipynb: train and test the model

To train a new model: change the parameters in the Train_test.ipynb

To test the existing model: run the last two cells of Train_test.ipynb

Packages

Project uses Python:

Python version: 3.6.12
Numpy version: 1.19.1
Matplotlib version: 3.3.1
OpenAI gym version: 0.17.3
Tensorflow=2.0.0
Tensorboard=2.3.0