Pytorch for my project

load image

data show

Run

Enter the following command in the current directory：

python -m visdom.server

run at train.py

Test Result