pytorch【迭代训练模块之可视化】三、TensorBoard使用(二)

(代码见lesson-22)

一、add_image 和 make_grid

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

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

你可能感兴趣的:(pytorch【迭代训练模块之可视化】三、TensorBoard使用(二))