pytorch入门
###_###_###
这个作者很懒,什么都没留下…
展开
-
RuntimeError:Input type(torch.FloatTensor)and weight type (torch.cuda.FloatTensor)should be the same
模型中输入为cuda类型的数据,调用模型时需更改原创 2021-11-08 19:46:17 · 250 阅读 · 0 评论 -
pytorch
TensorBoard使用(我理解是图像显示) from torch.utils.tensorboard import SummaryWriter from PIL import Image import numpy as np writer=SummaryWriter("logs") # 创建日志文件夹 writer.add_image("test",img_array,2,dataformats='HWC') # table ,input,索引 dataformats在数据类型转化原创 2021-11-06 14:04:58 · 841 阅读 · 0 评论