VisualDL

class paddle.callbacks.VisualDL ( log_dir )

VisualDL 是一个visualdl( 飞桨可视化分析工具 )的回调类。该类将训练过程中的损失值和评价指标储存至日志文件中后，启动面板即可查看可视化结果。

参数：

  • log_dir (str) - 输出日志保存的路径。

代码示例：

  1. import paddle
  2. import paddle.vision.transforms as T
  3. from paddle.static import InputSpec
  5. inputs = [InputSpec([-1, 1, 28, 28], 'float32', 'image')]
  6. labels = [InputSpec([None, 1], 'int64', 'label')]
  8. transform = T.Compose([
  9.     T.Transpose(),
  10.     T.Normalize([127.5], [127.5])
  11. ])
  12. train_dataset = paddle.vision.datasets.MNIST(mode='train', transform=transform)
  13. eval_dataset = paddle.vision.datasets.MNIST(mode='test', transform=transform)
  15. net = paddle.vision.LeNet()
  16. model = paddle.Model(net, inputs, labels)
  18. optim = paddle.optimizer.Adam(0.001, parameters=net.parameters())
  19. model.prepare(optimizer=optim,
  20.             loss=paddle.nn.CrossEntropyLoss(),
  21.             metrics=paddle.metric.Accuracy())
  23. ## uncomment following lines to fit model with visualdl callback function
  24. # callback = paddle.callbacks.VisualDL(log_dir='visualdl_log_dir')
  25. # model.fit(train_dataset, eval_dataset, batch_size=64, callbacks=callback)