insightface mxnet训练 out of Memory

11年it研发经验，从一个会计转行为算法工程师，学过C#，c++，java，android，php，go，js，python，CNN神经网络，四千多篇博文，三千多篇原创，只为与你分享，共同成长，一起进步，关注我，给你分享更多干货知识!

发现menet训练分配gpu代码：

    cvd ="4,5"# os.environ['CUDA_VISIBLE_DEVICES'].strip()if len(cvd)>0:for i in range(len(cvd.split(','))):ctx.append(mx.gpu(i))if len(ctx)==0:ctx = [mx.cpu()]print('use cpu')else:print('gpu num:', len(ctx),cvd)prefix = args.prefix

每次从gpu0开始训练，当再开训练时，0gpu已经被占用，所以就报错了。

正确代码：

 cvd ="0,1"# os.environ['CUDA_VISIBLE_DEVICES'].strip()if len(cvd)>0:for i in cvd.split(','):ctx.append(mx.gpu(int(i)))if len(ctx)==0:ctx = [mx.cpu()]print('use cpu')else:print('gpu num:', len(ctx))

insightface mxnet训练 out of Memory相关推荐

insightface mxnet训练horovod版
依赖项: horovod Horovod是Uber开源的又一个深度学习工具,它的发展吸取了Facebook "Training ImageNet In 1 Hour" 与百度 &q ...
insightface mxnet训练旧版
11年it研发经验,从一个会计转行为算法工程师,学过C#,c++,java,android,php,go,js,python,CNN神经网络,四千多篇博文,三千多篇原创,只为与你分享,共同成长,一起进 ...
mxnet 训练--如何生成rec 数据 +自己在本机测试的结果
mxnet 训练: http://www.th7.cn/system/lin/201512/145949.shtml 一.MXNet,不是直接处理原生数据比如图片之类的,而是处理的自定义的一种文件格式 ...
Mxnet训练自己的数据集并测试
使用Mxnet训练图片分类类器 1.准备数据: (1)建立一个根目录然后,再为每一个类别的图片建立一个子文件夹,把每一类图片放入对应的子文件夹即可. --root: ----class1 ----cl ...
insightface人脸训练数据rec文件制作
MXNet框架用于做图像相关的项目时,读取图像主要有两种方式:第一种是读.rec格式的文件,优点是.rec文件比较稳定,移植到别的电脑上也能复现,缺点是占空间(.rec文件的大小基本上和图像的存储大小 ...
mxnet训练arcface加速实验
本地训练服务器:Ubuntu16.04.cuda10.1.cudnn7.6.5.1080Ti x 8 主要是从 https://github.com/deepinsight/insightface/i ...
人脸识别之insightface开源代码使用：训练、验证、测试（2）
码字不易,转载请注明出处! 有了前面的准备工作之后,我们就开始动手了. Let The Hunt Begin! 数据集规模数据当然是越多越好,然而实际我们可能没有那么多数据,那么多大的量就可以了呢? ...
基于Insightface的人脸识别
Insightface是在人脸识别方向继facenet:https://github.com/davidsandberg/facenet后又一个比较完善的开源项目. 先给出论文地址:https://a ...
insightface 人脸识别
https://blog.csdn.net/weixin_43013761/article/details/99646731 接下来,我会为大家解析insightFace的每一个要点,从论文复现,源码 ...

insightface mxnet训练 out of Memory

insightface mxnet训练 out of Memory相关推荐

最新文章

热门文章