报错:logger = torch.utils.tensorboard.SummaryWriter(ckpt_dir=settings.ckpt_dir)
该报错是在执行模型训练时碰到,代码中报错行:logger = SummaryWriter(log_dir=settings.ckpt_dir)
问题原因:这类问题报错通常是tensorboard版本原因,可能高版本支持但低版本无此参数
解决方法:使用conda install安装新版tensorboard
conda install -c conda-forge tensorboard
补充:
torch.utils.tensorboard.writer.SummaryWriter(log_dir=None, comment=''
, purge_step=None, max_queue=10, flush_secs=120, filename_suffix='')
def __init__(self, logdir):
try:
from torch.utils.tensorboard import SummaryWriter
self.writer = SummaryWriter(log_dir=logdir)
报错
问题原因:np.array可使用 shape;而对于列表list,却不能使用shape来查看列表的维度
解决方法:np.array(list)转换后使用
类似的报错:list indices must be integers or slices, not tuple
报错
问题原因:tensor的数据类型不对
解决方法:
x=x.to(torch.float32)
labels = labels.long()
【参考文章】
circular import问题解方案
列表取片操作易错点
tensorboard.writer.SummaryWriter
tensorboard版本问题
tensorboard.writer.SummaryWriter方法示例