This commit is contained in:
JohnJim0816
2021-03-15 17:16:36 +08:00
parent 8d06642c56
commit 66fd8ef018
7 changed files with 255 additions and 0 deletions

35
codes/DQN/README.md Normal file
View File

@@ -0,0 +1,35 @@
## 思路
见[我的博客](https://blog.csdn.net/JohnJim0/article/details/109557173)
## 环境
python 3.7.9
pytorch 1.6.0
tensorboard 2.3.0
torchvision 0.7.0
## 使用
train:
```python
python main.py
```
eval:
```python
python main.py --train 0
```
可视化:
```python
tensorboard --logdir logs
```
## Torch知识
[with torch.no_grad()](https://www.jianshu.com/p/1cea017f5d11)