from tensorboardX import SummaryWriter
writer = SummaryWriter()
这里的SummaryWriter如果不写参数的话,会默认在当前目录下的runs文件夹下生成日志。
在验证的代码部分加入:
writer.add_scalar("test_accuracy", accuracy, train_step)
为了不使最后的结果生成混乱,要在结束的地方加
writer.close()
from tensorboardX import SummaryWriter
writer = SummaryWriter()
这里的SummaryWriter如果不写参数的话,会默认在当前目录下的runs文件夹下生成日志。
在验证的代码部分加入:
writer.add_scalar("test_accuracy", accuracy, train_step)
为了不使最后的结果生成混乱,要在结束的地方加
writer.close()