- 博客(4)
- 收藏
- 关注
RSS订阅原创 多维张量中取第二大值
取多维度张量中的第二大值：先将最大值置零：x[x==max]=0，x，max两者维度需相同在用max函数取剩下的最大值x=torch.max(x,dim=1)[0]
2021-03-10 16:59:03
2073
原创 Linux开机进入救援模式怎么解决
Linux系统（ubuntu18.04版本），进入救援模式，出现以下这种情况：方法：①输入自己的root密码进shell②vim /etc/fstab查看一下这个文件（/etc/fstab这个配置文件是管理开机自动挂载的一个文件）③看看/etc/fstab中是不是有问题设备我的解决办法是如图所示。...
2021-03-09 16:18:23
6398
2
原创 针对神经网络训练过程中显存爆炸的问题
针对神经网络训练过程中显存爆炸的问题在神经网络中，经常受到硬件的限制而导致在训练过程中out of memory，所以在训练过程中适当降低显存是必要的，方法是在一次训练结束后删除无用的数据，并用torch.cuda.empty_cache释放显存空间。例：del a,torch.cuda.empty_cache但是相对于网络的庞大参数，此方法只能降低有限的显存空间。经过试验，加了此代码之后显存只减少2M（在batch_size（4,16）之下测试，总10789降至10787）；删除了concat之
2020-09-24 09:58:46
1467
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人