安裝
指定GPU訓(xùn)練
CUDA_VISIBLE_DEVICES=1 python main.py
tensorboard
tensorboard --logdir ./stargan/logs
print( loss.data[0])
報(bào)錯(cuò)：
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
修改：
loss.data[]
改為：
loss.item()
更新pytorch 到 pytorch 1.7-cuda11
# 卸載舊版本
ssh 18.18.10.201
pip uninstall torch
# 安裝新版本
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
