easy-rl/codes/dqn

思路

见我的博客

环境

python 3.7.9

pytorch 1.6.0

tensorboard 2.3.0

torchvision 0.7.0

使用

train:

python main.py 

eval:

python main.py --train 0 

可视化：

tensorboard --logdir logs 

Torch知识

with torch.no_grad()