小土堆Pytorch学习笔记（三、TensorBoard的使用（1））

一、tensorboard_1

from torch.utils.tensorboard import SummaryWriter
查看一下SummaryWriter，如下：

class SummaryWriter(object):
    """Writes entries directly to event files in the log_dir to be
    consumed by TensorBoard.

    The `SummaryWriter` class provides a high-level API to create an event file
    in a given directory and add summaries and events to it. The class updates the
    file contents asynchronously. This allows a training program to call methods
    to add data to the file directly from the training loop, without slowing down
    training.
    """

查看一下add_scaler

   def add_scalar(
        self,
        tag,
        scalar_value,
        global_step=None,
        walltime=None,
        new_style=False,
        double_precision=False,
    ):
        """Add scalar data to summary.

        Args:
            tag (string): Data identifier
            scalar_value (float or string/blobname): Value to save
            global_step (int): Global step value to record
            walltime (float): Optional override default walltime (time.time())
              with seconds after epoch of event
            new_style (boolean): Whether to use new style (tensor field) or old
              style (simple_value field). New style could lead to faster data loading.

创建一个实例：

from torch.utils.tensorboard import SummaryWriter
# 创建一个实例
writer = SummaryWriter("logs")
# 再使用两个类方法
# writer.add_image()

for i in range(100):

    writer.add_scalar("y=2x", 2*i, i)

writer.close()

在终端打开tensorboard查看生成的文件
tensorboard --logdir = logs
logdir = 事件文件所在文件夹名
笔者在pycharm打开本地终端时是PS端，而小土堆打开的直接是虚拟环境里的终端，可以去看小土堆配置pytorch的视频可以看到他改变了pycharm的终端，他使用的是虚拟环境里的anaconda prompt的cmd.exe， 可以在文档——设置——查找终端——进行修改。具体做法;打开anaconda prompt的文件地址，如果看到快捷方式的话继续打开文件地址，右键属性——安全——对象名称——复制下来，再粘贴到pycharm——文件——设置——终端——shell路径，进行更改。如果还不行，其实可以直接在终端里选择命令行终端（command prompt)，输入指令也可以使用tensorboard。

(xxxpytorch) F:\learn_pytorch>tensorboard --logdir=logs
TensorFlow installation not found - running with reduced feature set.
Serving TensorBoard on localhost; to expose to the network, use a proxy or pass --bind_all
TensorBoard 2.12.2 at http://localhost:6006/ (Press CTRL+C to quit)
#可以更改端口，防止端口已经被别人占用,如下：
tensorboard --logdir=logs --port=6007

笔者运行的结果如下：

跟小土堆的结果有点差别。