最新 最热

基于Pytorch实现的快速人脸识别模型

本项目参考了ArcFace的损失函数结合MobileNet,意在开发一个模型较小,但识别准确率较高且推理速度快的一种人脸识别项目,该项目训练数据使用emore数据集,一共有85742个人,共5822653张图片,使用lfw-align-128数据集作为测试数...

2021-12-07
0

【pytorch】固定(freeze)住部分网络

因为:即使对bn设置了 requires_grad = False ,一旦 model.train() ,bn还是会偷偷开启update( model.eval()模式下就又停止update )。(详见【pytorch】bn) 所以:train每个epoch之前都要统一重新定义一下这块,否则容易出问题。...

2021-12-06
0

【pytorch】onnx

pytorch任意形式的model(.t7、.pth等等)转.onnx全都可以采用固定格式。

2021-12-06
1

【pytorch】tnn

首先确保已经安装了tnn环境:onnx2tnn转tnn:def onnx2tnn(self): assert os.path.isfile(simplified_onnx_path) os.system('cd {}n' 'python onnx2tnn.py {} ...

2021-12-06
0

《自然语言处理实战入门》第二章:NLP 前置技术(深度学习) ---- pytorch

An open source machine learning framework that accelerates the path from research prototyping to production deployment.

2021-12-06
1

自定义Gym环境之井字棋

在文章 中 ,我们了解了一个简单的环境应该如何定义,并使用 print 简单地呈现了环境。在本文中,我们将学习自定义一个稍微复杂一点的环境——井字棋。回想一下井字棋游戏:...

2021-12-06
1

Meta 开移动端 AI 生成神器 PyTorch Live,打造人工智能驱动的移动体验

近日,PyTorch 开发者大会如期召开。在会上,Meta 发布了PyTorch Live,这是一套可以为移动端用户提供人工智能体验的工具。PyTorch Live 支持单一的编程语言JavaScript,可以为 Android 和 iOS 两个移动端操作系统开发的应用...

2021-12-04
0

[源码解析] PyTorch 分布式 Autograd (4) ---- 如何切入引擎

上文我们看到了AutogradMetadata,DistAutogradContainer 和 DistAutogradContext 等一系列基础类。我们知道了分布式autograd如何基于RPC进行传递,如何在节点之间交互,节点如何区分维护这些Session。本文继续分析,主要目...

2021-12-04
1

如何参与PyTorch社区开源贡献?

作为深度学习框架OneFlow的一个全职开发者(声明我是菜鸡),基本上每天都会和PyTorch打交道,所以自问自答一下这个问题,希望对想为开源项目做贡献的小伙伴们提供一些思路也希望这个问题能获得更多回复。...

2021-12-02
0

from torch._C import *ImportError: DLL load failed 报错解决方法 亲测可用

from torch._C import * ImportError: DLL load failed: 找不到指定的程序。

2021-12-01
0