These programs rely mainly on pytorch, gym and optuna.
They can be ran in two ways:
-
python main.pywill train a simple convolutional DDQN model (seeDQN.py). -
The
experiments.ipynbnotebook will let you run hyper-parameter tuning experiments on our full model (seeDQN_parametrized.py).