思路
见我的博客
环境
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
见我的博客
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