- 博客(5)
- 收藏
- 关注
原创 查看GPU占用率的方法以及解决神经网络训练过程中GPU占用率低的问题
一、查看GPU占用率的方法1. 使用终端命令nvidia-smi -l 3查看GPU使用情况。其中命令末尾的3表示3秒刷新一次,时间可自行修改。2. 使用任务管理器查看。GPU下方的选项需选择cuda才可以查看具体的GPU占用率。二、GPU占用率过低可以根据自己显存的大小来调节batchsize和num_workers参数博主这里通过调大workers以及batchsize的大小很好地提高了GPU的占用率,但是CPU的占用率依然是拉满,这是因为在训练神经网络的过程中在CPU主要负责对数
2022-03-26 11:09:53
29740
8
原创 SE block论文地址、详细代码及网络解析
论文:Squeeze-and-Excitation Networks链接:https://arxiv.org/pdf/1709.01507.pdf该篇论文是ImageNet最后一届比赛的冠军作品,在很多经典网络结构中都可以看到它的身影,例如Mobilenet v3。其实是一种通道注意力机制。由于特征压缩和FC的存在,其捕获的通道注意力特征是具有全局信息的。该论文提出了一种新的结构单元——“Squeeze-and Excitation(SE)”模块,可以自适应的调整各通道的特征响应值,对通道间的内部依赖
2022-02-24 22:32:26
1699
2
原创 解决ValueError: Expected input batch_size () to match target batch_size ().问题
一般这种问题是由于输入图像的大小不匹配导致,class Network(nn.Module): def __init__(self, height_size, width_size, num_classes): super(Network, self).__init__() self.height_size = height_size self.width_size = width_size self.conv2d_1 = nn.Co
2022-02-11 18:22:53
28894
2
原创 利用python将二进制图片转化成图片并展示
mnist数据集的数据以二进制格式存储,下面读取mnist数据集的指定图片并展示import torch.optimfrom network import Mnist_networkimport matplotlib.pyplot as pltimport numpy as npfrom PIL import Imagemnist_network = Mnist_network()mnist_network = mnist_network.load_state_dict(torch.load
2022-02-09 21:04:25
5762
原创 解决pycharm运行程序时出现Run ‘pytest in XXX.py‘的问题
一、出现的问题右键运行时和正常运行不一致,只执行单一函数并且出现以下画面二、解决方法进入到File->Settings->Tools->Python integrated Tools界面并将Autodetect(pytest)设置为Unittests即可解决该问题。...
2022-02-09 16:33:47
2562
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人