2021年12月24日
numpy—array类型 与 pytorch—tensor类型 互相转换
摘要： 一、numpy_array 转 torch_tensor import torch torch_data = torch.from_numpy(numpy_data) 二、torch_tensor 转 numpy_array 1 、 numpy_data = torch_data.numpy() 2 阅读全文
posted @ 2021-12-24 13:08 微笑黎 阅读(812) 评论(0) 推荐(0)
pytorch 保存模型 (.pt, .pth, .pkl)
摘要： .pt, .pth, .pkl 仅仅是后缀不同 一、只保存模型权重参数，不保存模型结构 保存： torch.save(model.state_dict(), 'mymodel.pth') 加载： torch.load('mymodel.pth路径') #加载后是一个OrderedDict类型 调用： 阅读全文
posted @ 2021-12-24 11:37 微笑黎 阅读(1161) 评论(0) 推荐(0)
.mat文件读取 .npy .npz文件保存
摘要： 一、利用scipy.io模块 二、io.loadmat方法读取 读取到的.mat文件是一个字典 可以通过matr.keys()查看keys 三、保存为.npy/.npz 1、.npy .npy文件 通过 np.load()方法加载 格式为array 2、.npz train_x = train_x 阅读全文
posted @ 2021-12-24 10:38 微笑黎 阅读(872) 评论(0) 推荐(0)