使用
在train文件中加入
import torch
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter(log_dir="/project/train/tensorboard")
log_dir:一般不用特别指定它会自动写入.run里
环境配置参考文档
报错:module ‘distutils’ has no attribute ‘version’
pip install setuptools==59.5.0