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