07 2020 档案

pytorch中tensorboard的使用
摘要：导入包。 from torch.utils.tensorboard import SummaryWriter 实例化一个summaryWriter，设置记录保存在runs文件夹里。 writer = SummaryWriter('runs') 在运行中记录loss：比如每20个batch记录一次损失 阅读全文
posted @ 2020-07-24 10:57 肃木易 阅读(3215) 评论(0) 推荐(1) 编辑