CheXNet-master：CUDA out of memery

在 for i, (inp, target) in enumerate(test_loader): 前添加 with torch.no_grad(): 就不报错了

with torch.no_grad():for i, (inp, target) in enumerate(test_loader):target = target.cuda()gt = torch.cat((gt, target), 0)bs, n_crops, c, h, w = inp.size()input_var = torch.autograd.Variable(inp.view(-1, c, h, w).cuda())output = model(input_var)output_mean = output.view(bs, n_crops, -1).mean(1)pred = torch.cat((pred, output_mean.data), 0)

但是效果不佳：

The average AUROC is 0.531
The AUROC of Atelectasis is 0.552835646499197
The AUROC of Cardiomegaly is 0.4891981635698571
The AUROC of Effusion is 0.5078801344734772
The AUROC of Infiltration is 0.5471494224277326
The AUROC of Mass is 0.5019205110036506
The AUROC of Nodule is 0.48564796421763534
The AUROC of Pneumonia is 0.49787587924670523
The AUROC of Pneumothorax is 0.46888930706822896
The AUROC of Consolidation is 0.45797680791836254
The AUROC of Edema is 0.5500617972215441
The AUROC of Emphysema is 0.7189693346796524
The AUROC of Fibrosis is 0.548014619318718
The AUROC of Pleural_Thickening is 0.47339353023337755
The AUROC of Hernia is 0.638215609588036

CheXNet-master：CUDA out of memery相关推荐

BEP 7：CUDA外部内存管理插件（下）
BEP 7:CUDA外部内存管理插件(下) Numba依赖向库中添加EMM插件的实现自然会使Numba成为库的依赖项,而以前可能没有.为了使依赖关系可选,如果需要的话,可以有条件地实例化并注册EMM ...
BEP 7：CUDA外部内存管理插件（上）
BEP 7:CUDA外部内存管理插件(上) 背景和目标在CUDA阵列接口使得能够共享不同的Python之间的数据库的访问CUDA设备.但是,每个库都与其它库区别对待.例如: • Numba在内部管理 ...
CUDA刷新器：CUDA编程模型
CUDA刷新器:CUDA编程模型 CUDA Refresher: The CUDA Programming Model CUDA,CUDA刷新器,并行编程这是CUDA更新系列的第四篇文章,它的目标是 ...
英伟达官宣：CUDA 将不再支持 macOS
点击上方"AI遇见机器学习",选择"星标"公众号原创干货,第一时间送达内容提要:或许,今后我们再也看不到搭载英伟达显卡的新款苹果电脑了.英伟达在最近的一份说 ...
yolov配置之：cuda、 cudnn安装
cuda. cudnn安装版本参数配置 1.介绍 2.tensorflow的版本的选择 3.CUDA的安装, 4.cuDNN安装 5.环境配置 6.测试常见问题 Could not find No ...
CUDA 编程上手指南：CUDA C 编程及 GPU 基本知识
作者丨科技猛兽编辑丨极市平台本文原创首发于极市平台,转载请获得授权并标明出处. 推荐大家关注极市平台公众号,每天都会更新最新的计算机视觉论文解读.综述盘点.调参攻略.面试经验等干货~ 目录 1 C ...
cuda编程_CUDA刷新器：CUDA编程模型
CUDA刷新器:CUDA编程模型 CUDA Refresher: The CUDA Programming Model CUDA,CUDA刷新器,并行编程这是CUDA更新系列的第四篇文章,它的目标是 ...
NVIDIA CUDA 高度并行处理器编程（一）：CUDA简介
NVIDIA CUDA 高度并行处理器编程(一):CUDA简介 1. 数据并行性 2. CUDA的程序结构 3. 向量加法kernel函数 4. 设备全局存储器与数据传输 5. kernel 函数与线 ...
CUDA：CUDA内核中内联PTX（汇编语言）用法的实例
CUDA:CUDA内核中内联PTX(汇编语言)用法的实例 // System includes #include <stdio.h> #include <assert.h>// ...
CUDA：CUDA转OPENGL的实例
CUDA:CUDA转OPENGL的实例 #include <helper_cuda.h>// clamp x to range [a, b] __device__ float clamp( ...

CheXNet-master：CUDA out of memery

CheXNet-master：CUDA out of memery相关推荐

最新文章

热门文章