![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Deep Learning
星辰大海在梦中
这个作者很懒,什么都没留下…
展开
-
subprocess.CalledProcessError: Command ‘[‘/home/bjt/miniconda3/envs/HeadDetect/bin/python‘, ‘-
训练segformer时出现的,改配置文件‘‘/SegFormer-master/local_configs/segformer/B5/segformer.b5.640x640.ade.160k.py’’的num_classes,由默认的150改为自己的类别个数即可。原创 2022-07-12 18:20:43 · 721 阅读 · 1 评论 -
深度学习知识蒸馏、量化、剪枝学习笔记
1.网络最后一层(全连接层)的输出为类别得分scores(也加logits),使用sofxmax对logits映射为概率分布,且和为1,添加t参数,使网络概率分布更加平缓.如下为添加t参数方式.import numpy as np#原softmax函数,numpy实现def softmax(x): x_exp = np.exp(x) return x_exp / np.sum(x_exp)output = np.array([0.1, 1.6, 3.6])print(sof原创 2022-01-01 21:12:02 · 1868 阅读 · 0 评论 -
深度学习总结
梯度爆炸解决方法:梯度裁剪梯度消失解决方法:LSTM、GRU原创 2018-12-05 16:34:24 · 224 阅读 · 0 评论 -
tensorflow 训练时出现failed to allocate 18.41M (19300352 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY
显存满了,nvidia-smi查看占用显存的程序,在下方显示占用现存的程序,sudo kill -9 PID(占用程序的ID)即可。原创 2019-03-07 21:33:02 · 3904 阅读 · 3 评论 -
LSTM
原创 2019-04-23 20:42:18 · 166 阅读 · 0 评论