![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
追风的Zoom
中科院计算所直博生,研究方向为Processing in Memory等。
展开
-
vscode使用jupyter切换虚拟环境报错“IPyKernel 没有安装到解释器”
今天想在vscode中的jupyter使用虚拟环境的内核,在jupyter右上角点击切换后,运行代码显示该环境需要安装或者更新ipykernel。问题出在点击安装之后,一切安装顺利,在命令行里运行该虚拟环境也看得到ipykernel的包,可是再次运行代码,仍然现实该环境需要安装或者更新ipykernel。期间发现两个问题:当前jupyter显示虚拟环境的python小版本(3.7.10)与命令行中虚拟环境的不一致(3.7.11),对vscode中的jupyter卸载后重新安装,解决;版本一致后仍然原创 2021-09-30 15:15:50 · 16738 阅读 · 10 评论 -
load预训练模型后,测试准确率很低问题自查
今天遇到了这样一个问题,做神经网络量化的过程中,load训练好的高精度模型作为pretrain model,然后立即test accuracy为10%,以下为几个可能出现问题的点:查看load模型的路径有无问题查看加载的数据集有无问题(有时会出现用svhn数据集去验证cifar10数据集的事件)查看load模型当时在训练时的acc:info = torch.load('modelpath')info.keys() # 列出保存checkpoints时的所有参数,包括best_accinfo[原创 2021-06-15 14:50:27 · 2703 阅读 · 0 评论 -
Pytorch下用GPU加速数据处理
尽量使用Pytorch提供的函数,以分块为例,使用split()函数进行分割,而不要按照最原始的思路element by element的遍历:原创 2020-04-06 16:32:11 · 504 阅读 · 0 评论 -
cuda8.0环境下安装DGL
在DGLv0.3及之后的版本中,将cpu和gpu版本分开了,而在安装gpu版本时仅支持cuda9.0和10.0。因此可以选择安装DGLv2.0:pip install dgl==0.2原创 2020-04-03 10:46:58 · 1970 阅读 · 2 评论 -
记录Pytorch中遇到的问题
在feature层到classifier层中,若出现如下错误:则需要改变FC层的入口参数,如此图中可以改为:x = x.view(out.size(0), -1)self.linear = nn.Linear(320, 10) # 320为入口参数...原创 2019-03-01 15:03:35 · 14820 阅读 · 6 评论 -
关于opencv
一行在window系统下安装:cmd命令行中输入pip install opencv-python (前置需要安装pip和numpy)原创 2019-03-30 15:11:14 · 172 阅读 · 0 评论