pytorch
文章平均质量分 50
qq_35407318
模型压缩和加速,计算机视觉
展开
-
记录pytorch错误
1错误解决方法:添加:map_location=‘cpu’如下torch.load(‘model’, map_location=‘cpu’)2错误解决方法添加: model = torch.nn.DataParallel(model)3错误解决方法不要使用.cuda() 或者.to(Drive)原创 2021-03-18 19:17:34 · 105 阅读 · 0 评论 -
深度学习的即插即用-trick笔记
trick目录1. Dynamic Convolution1. Dynamic Convolutionpaper的地址:https://link.zhihu.com/?target=https%3A//arxiv.org/pdf/1912.03458.pdf代码:https://github.com/pinery-sls/Dynamic-convolution-Pytorch????可以替换常规卷积层参考:https://zhuanlan.zhihu.com/p/208519425...原创 2021-03-13 23:07:07 · 832 阅读 · 0 评论 -
Yolov3训练时报错RuntimeError: CUDA error: device-side assert triggered
报错Traceback (most recent call last): File "train.py", line 122, in <module> loss, outputs = model(imgs, targets) File "/home/pinery/anaconda3/envs/tf3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 532, in __call__ result原创 2020-12-09 19:17:54 · 666 阅读 · 0 评论 -
pytorch得到模型的计算量和参数量
文章目录方法1方法2方法3方法4方法1pytorch自带方法,计算模型参数总量 total = sum([param.nelement() for param in model.parameters()]) print("Number of parameter: %.2fM" % (total/1e6))方法2计算模型参数总量和模型计算量def count_params(model, input_size=224): # param_sum = 0 with open('原创 2020-10-29 14:38:14 · 17694 阅读 · 2 评论 -
pytorch出现的警告
遇到的警告:UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python numbertest_loss += F.nll_loss(output, target, size_average=False).data[0]解决方法:把data[0]改为item()test_loss +原创 2020-10-10 19:59:39 · 8864 阅读 · 6 评论