pytorch
pytorch使用记录
吴天德少侠
这个作者很懒,什么都没留下…
展开
-
libtorch学习第八
基于c++17。原创 2024-01-24 13:05:48 · 362 阅读 · 0 评论 -
libtorch学习第六
【代码】libtorch学习第六。原创 2024-01-23 14:35:06 · 370 阅读 · 0 评论 -
libtorch学习第五
【代码】libtorch学习第五。原创 2024-01-23 13:27:01 · 339 阅读 · 0 评论 -
libtorch学习第四
【代码】libtorch学习第四。原创 2024-01-23 13:16:15 · 329 阅读 · 0 评论 -
libtorch学习第三
【代码】libtorch学习第三。原创 2024-01-23 11:36:37 · 352 阅读 · 0 评论 -
libtorch学习第二
估计大多数人都是第3种用的多。原创 2024-01-23 11:22:04 · 373 阅读 · 0 评论 -
libtorch学习第一
max。原创 2024-01-23 10:55:10 · 384 阅读 · 0 评论 -
使用torch求函数参数
经过标注人员测试发现一组对应的。,跟我们用来生成数据的函数。原创 2024-01-22 09:18:33 · 770 阅读 · 0 评论 -
使用torch的自动微分实现自定义函数优化
目的是利用torch已经有的自动微分机制,进行参数迭代更新,就不用自己写代码算了。原创 2023-06-16 10:21:23 · 336 阅读 · 0 评论 -
torch迁移学习,冻结层
2. 打印参数名称输出:3. 冻结最后的out层4. 打印没有冻结的层此时应该只有层原创 2023-05-05 11:07:30 · 155 阅读 · 0 评论 -
pytorch的安装方法
后面的内容,例如指定。原创 2023-04-18 10:50:22 · 192 阅读 · 0 评论 -
生成monai框架的dataset的json文件
在使用导入数据,以及时,使用json文件组织数据。那么假设我们的数据图像文件放在images文件夹里面,标签数据放在labels文件夹里面,如何生成这种格式的json文件呢,并且训练集和验证集的比例为8:2。原创 2023-03-27 15:37:36 · 262 阅读 · 0 评论 -
pytorch的summary和画网络结构图
在里面,可以直接调用和查看和保存网络的结构。但是在torch里面没有这么简单。需要借助额外的包和torchviz。原创 2023-03-21 13:13:33 · 467 阅读 · 0 评论 -
swin unetr的3D语义分割
3d医学图像分割原创 2023-02-15 09:17:05 · 2938 阅读 · 32 评论 -
pytorch实现自定义医学3d图像数据集dataset
pytorch原创 2023-01-31 16:32:44 · 753 阅读 · 0 评论 -
pytorch单机多卡训练的例子
pytorch原创 2022-12-05 16:56:17 · 483 阅读 · 0 评论 -
pytorch的gather
1. 官网解释out[i][j][k]=input[[index[i][j][k]][j][k], if dim==0out[i][j][k] = input[[index[i][j][k]][j][k],\ if \ dim==0out[i][j][k]=input[[index[i][j][k]][j][k], if dim==0以此类推意思就是说,假设我要求out[i][j]out[i][j]out[i][j]位置的值,如果dim=0,那么i就会被inde原创 2022-04-03 20:31:22 · 1957 阅读 · 0 评论