pytorch【迭代训练模块之可视化】三、TensorBoard

2020-02-24  本文已影响0人  米线织毛衣
(代码见lesson-22)

一、add_image 和 make_grid

这个函数可以可视化图像。

注意尺度,如果输入的像素值全都是0-1,那么会乘255再显示。如果输入像素值在0-255,会直接显示。

上一篇 下一篇

猜你喜欢

热点阅读