解决报错
SummaryWriter
是 PyTorch 中用于将数据写入 TensorBoard 日志文件的工具。它可以记录训练过程中的损失、准确率、模型参数等信息,并可视化这些信息以便于分析和监控模型的训练过程。
在训练模型的过程中遇到了如下的报错:
识别不出设置的文件路径。
几番波折,找到原因为 : 文件目录中含有中文,换成不包含中文的路径就不报错了
import os
from torch.utils.tensorboard import SummaryWriter
# 检查目录是否存在,如果不存在则创建
log_dir = "D:\log"
if not os.path.exists(log_dir):
os.makedirs(log_dir)
writer = SummaryWriter(log_dir)