参考链接小土堆pytorch教程:PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】
SummaryWriter类
- 主要使用SummaryWriter类来进行变量的保存,它可以向指定的文件夹写入事件文件,这个事件文件可以被tensorboard读取。这个类需要传入的主要参数为
log_dir
和comment
,前者设定事件文件的保存路径,后者可以在文件夹名称后添加一些备注,如学习率、batch_size等做一个区分。
保存需要可视化的事件文件
-
主要涉及的函数为
.add_scalar()
,通过这个函数可以传入图片的名称,纵轴数据以及横轴数据。对应的参数名分别为tag
、scalar_value
和global_step
。 -
注意事项:不同的图记得保存成不同的
tag
名,否则会错乱。并且写完要close。 -
如果报错
ImportError: TensorBoard logging requires TensorBoard with Python summary writer installed. This should be available in 1.14 or above.
直接pip install tensorboard
就可以,或者如果是版本太低则pip install --upgrade tensorb