![](https://img-blog.csdnimg.cn/20201014180756923.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
机器学习与深度学习
本专栏用于介绍和总结本人学习机器学习和深度学习过程中的内容
牧之原
努力成为一名优秀的算法工程师!
展开
-
Faiss 安装与配置(最新)
Faiss 简介Faiss 是一个用于高效相似性搜索和密集向量聚类的库。它包含在任意大小的向量组中搜索的算法,直到可能不适合 RAM 的向量组。它还包含用于评估和参数调整的支持代码。Faiss 是用 C++ 编写的,带有完整的 Python 包装器(版本 2.0+ 和 3.0+)。一些最有用的算法是在 GPU 上实现的。它由Facebook AI Research开发。相关链接:1: faiss官方文档2: github地址开始1. 安装faiss推荐使用Anaconda安装,如果你的电原创 2021-08-11 20:09:01 · 23351 阅读 · 1 评论 -
CPU和GPU模型以及模型参数之间的转化
模型转化 # GPU模型 -> GPU参数 / CPU模型 -> CPU参数 model.load_state_dict(torch.load('outputs/epoch4.pt')) # GPU模型 -> CPU 参数 model.load_state_dict(torch.load('outputs/epoch4.pt', map_location=lambda storage, loc: storage.cuda)) #原创 2021-04-28 21:39:31 · 598 阅读 · 0 评论 -
Pytorch搭建神经网络入门
简介Pytorch作为当前最受欢迎的深度学习框架之一,是笔者十分推荐大家学习和使用的。以下代码,我使用Pytorch提供的线性函数简单搭建了一个神经网络,用以实现对一次函数y = 2*x + 1的拟合,用以帮助大家实现从 0 到 1 搭建神经网络模型。代码以下为详细代码:import torchimport torch.nn as nnimport torch.optim as optimimport numpy as np# 搭建神经网络模型class Model(nn.Modul原创 2021-04-07 19:05:48 · 357 阅读 · 0 评论 -
Pytorch保存训练模型
简介我们在使用Pytorch深度学习框架训练自己的模型时,需要保存训练结果以便于下次直接使用训练好的模型对新数据进行预测。本文将介绍Pytorch官方提供的两种模型保存方法内容方法1:仅保存模型参数代码如下:# 保存模型状态字典,PATH为保存的路径以及文件名# 例如: PATH='models/myModel.pt'torch.save(model.state_dict(), PATH)# 加载模型# 先实例化一个模型model = TheModelClass(...)# 加载状态原创 2021-04-06 20:23:25 · 747 阅读 · 0 评论