报错
Deserve_p
这个作者很懒,什么都没留下…
展开
-
kaggle 深度学习训练内存不够
在kaggle上训练VGG网络,为了实现一个可视化算法,运用了pytorch中的hook机制。但是在训练中出现了爆内存的现象,总结一下几种解决办法:无关的变量使用完要删掉import gcdel variablesgc.collect()使用hook机制,如果多次使用,一定要每次用完就删除掉,否则多次重复register_hook,会叠加,占用很多内存空间handle = model.register_forward_hook(fn)handle.remove()...原创 2022-04-29 20:32:50 · 5585 阅读 · 2 评论 -
记一次报错
【小白入门遇到错误手足无措的无力感~在创建了几个gluon虚拟环境后,在pycharm使用,结果不能编译。仅仅是 from mxnet import nd 就会报错:AttributeError: module ‘numpy’ has no attribute ‘histogram_bin_edges’ si…!!!上网查,有的类似问题说是起的文件名字和包里面的函数名(还是库)冲突,改了就好了。后来我就把另外一个有相同mxnet包的虚拟环境删掉了。。。结果好了。。。其实我还是不懂为啥错。。。.原创 2021-03-30 22:21:14 · 485 阅读 · 1 评论