This commit is contained in:
JohnJim0816
2021-03-23 16:10:11 +08:00
parent d4690c2058
commit bf0f2990cf
198 changed files with 1668 additions and 1545 deletions

View File

@@ -1,33 +0,0 @@
## 思路
见[博客](https://blog.csdn.net/JohnJim0/article/details/111552545)
## 环境
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
```