百度已收录

安装

Pytorch官网

指定GPU训练

CUDA_VISIBLE_DEVICES=1    python main.py

tensorboard

tensorboard  --logdir ./stargan/logs