参考链接: torch.utils.tensorboard.writer.SummaryWriter
使用方式
torch.utils.tensorboard.writer.SummaryWriter(
log_dir=None,
comment='',
purge_step=None,
max_queue=10,
flush_secs=120,
filename_suffix='')
from torch.utils.tensorboard import SummaryWriter
# create a summary writer with automatically generated folder name.
writer = SummaryWriter()
# folder location: runs\Dec10_22-22-11_chenxuqi-desktop
# create a summary writer using the specified folder name.
writer = SummaryWriter("my_experiment")
# folder location: my_experiment
# create a summary writer with comment appended.
writer = SummaryWriter(comment="LR_0.1_BATCH_16")
# folder location: runs\Dec10_22-22-14_chenxuqi-desktopLR_0.1_BATCH_16
运行结果: