MHaneferd / mxnet.gluon

Some Python scripts to test out the Mxnet gluon package for deep learning with Doom

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

mxnet.gluon

Some Python scripts that I have modified and extended to learn the Mxnet gluon package for deep learning, and Python coding itself.. Since there are not much mxnet.gluon files out there, I have found some Pytorch files that I have translated to gluon.

gluon-a2c.py
This file have a simple synchronized actor critic model (A2C) which train on a ascii environment with a target and a player that shoots. You can easily modify number of shots for each episode. And also the number of episodes. The environment size is also customizable.

gluon-a2c-vizdoom.py
This file also have a simple synchronized actor critic model (A2C) which train on a ViZDoom environment. The Doom environment need to be installed before running. The logic of the python file is the same as with gluon-a2c.py, but it uses Convolutional network on the Doom environment. It is quite easy to experiment and debug models with the mxnet.gluon block, so feel free to send me some input if you see that the model could be made better. The configuration of which Doom WAD file to use must be modified before running (As I cannot imagine you have the same path as I)

gluon-dqn-vizdoom.py
This file uses DQN network model with mxnet gluon. It is translated from Pytorch file. ( I have not invented the logic here, just made it work with mxnet.gluon ). It uses Doom (Vizdoom) environment. The configuration of which Doom WAD file to use must be modified before running (As I cannot imagine you have the same path as I)

gluon-a2c-lstm-ascii.py
This file is based on the gluon-a2c.py , but has an LSTM layer in addition in the model. It performs with a much better accuracy than the latter. The LSTM layer is added into the model with one for the initialize, and one in the forward statement. The shape of the input has been reduced to 3D instead of 4D as in gluon-a2c.py

About

Some Python scripts to test out the Mxnet gluon package for deep learning with Doom

License:GNU General Public License v3.0


Languages

Language:Python 100.0%