We have saved models to load so you can test them. https://drive.google.com/drive/folders/1BNGJst7KJ2ODBlODfxWUYlMHck05Sm-Q?usp=sharing
NOTE: The saved models need to be in the same directory with our code.
How to run our code:
train:
model_name can be: DQN, DDQN, Dueling_DQN, Rainbow_DQN
python PATH/train_DQN_models.py model_name gpu_number
python PATH/train_TD3_FORK.py gpu_number
test:
model_name can be: DQN, DDQN, Dueling_DQN, Rainbow_DQN, TD3_FORK
python PATH/test.py model_name gpu_number
all the models was trained and tested with 2080ti gpu.