m0_56015025的博客

模型中输入为cuda类型的数据，调用模型时需更改

TensorBoard使用（我理解是图像显示）from torch.utils.tensorboard import SummaryWriterfrom PIL import Imageimport numpy as npwriter=SummaryWriter("logs") # 创建日志文件夹writer.add_image("test",img_array,2,dataformats='HWC') # table ,input,索引dataformats在数据类型转化