To train a model:
$ python3 main.py or python main.py
To test simualation:
$ python simulator.py
To test network with simulator:
$ python Network_test.py
The model is defined in dqn_model.py
The algorithm is defined in dqn_learn.py
The running script and hyper-parameters are defined in main.py
related paper link https://www.nature.com/articles/nature14236.pdf
if you want to use tensorboard in pythorch(monitoring DNN)
- install tensorboardX
pip install tensorboardX
- isntall tensorflow
pip install tensorflow
- launch tensorboard at your directory
cd dqn_learning_code
tensorboard --logdir runs
- copy the address and paste the address at webbrowser