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