- 博客(8)
- 收藏
- 关注
原创 Pytorch预训练模型pre-trained Models以及部分修改
用torchvision.models.XXX加载模型时用参数pretrained=True好像可以运行,但是会报红,用weights没问题,下面举了两个例子。返回的是features和avgpool两层,也就是.children返回的是模型的所有顶层模块。网上找到的都是几年前的了,更新一下最近的用法和踩的坑。则是直接用 顶层模块[ ]才能直接取出。参数修改没有研究过,后续需要再来补坑。容器中的前 20 层。
2024-06-15 19:38:47
159
原创 window下配置pytorch,nvidia-smi有显示但是nvcc -v显示不是内部/外部命令
CUDA其实是分为2个部分。一部分是用户态的驱动里,一部分是运行时、库和工具。英伟达官方已经把驱动和CUDA匹配的版本打包到一起了安装了驱动,所以调用nvidia-smi,就会显示用户态的CUDA驱动版本。下比nvidia-smi显示的cuda version小的cuda toolkit.我的情况是笔电4060显卡,我的解决办法是在。找对应GPUs supported,然后再到。这个王方浩答主的回答解释的比较清楚了,
2024-04-08 17:28:04
452
原创 关于golang make chan的第二个参数
而开辟管道大小,taskqueue[ i ] = make(chan interface , x),x表示开辟了x大小的管道。工作池taskqueue = make([ ]chan interface, x),x表示开了x个chan。
2023-05-22 12:08:54
169
2
原创 关于binary.read
binary.read底层是用io.readfull实现的,多次调用该函数会依次读下去,而不是每次都从头开始,参考链接:关于io.readfull理解(但是这个例子中string.NewReader返回的reader是指针,不知道是不是因为这个原因,还有待继续扒):https://blog.csdn.net/huang_2016/article/details/124639763
2023-05-17 11:21:43
169
1
原创 golang appears multiple times in workspace
go1.18之后引入的workplace模式不支持在一个workplace下有多个同名文件,还会一片红,删掉go.work文件vscode还是报红,但是编译没问题。
2023-05-10 23:18:51
348
1
原创 maya ncloth初级学习(无法播放预览问题)
maya2022使用ncloth进行布料落在台面上效果,点击播放无法预览,通过"preferences"-"animation"-"Evaluation"将mode由parallel切换至DG解决。立个flag,之后要查下这几种mode区别。
2022-11-06 23:52:06
612
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人