pytorch
pytorch基础知识和语法,用于平时的积累
Taboo-
这个作者很懒,什么都没留下…
展开
-
深度学习(pytorch)经常使用的一些小技巧
1 可视化模型结构以及每一层的shape和参数量。安装 torchinfo 包pip install torchinfo -i https://pypi.tuna.tsinghua.edu.cn/simple写代码 from torchinfo import summary base_model = uniformer_small() print(summary(base_model, (1, 3, 8, 224, 224)))...原创 2022-05-21 17:17:50 · 223 阅读 · 0 评论 -
加载pytorch已有模型,修改最后分类头
加载pytorch已有模型 修改分类头原创 2022-04-11 19:52:51 · 2118 阅读 · 0 评论 -
GPU训练完之后转移到CPU测试
深度学习 :GPU训练之后转到CPU测试(pytorch)原创 2022-03-18 20:00:11 · 1484 阅读 · 0 评论 -
pytorch中的 .masked_fill_()函数
1 What?明确函数功能,掩码操作,用value掩盖mask中相对应的元素。.masked_fill_(mask,value)2 HOW?要怎么使用。a = torch.randn(5,6)x = [5,4,3,2,1]mask = torch.zeros(5,6,dtype=torch.float)for e_id, src_len in enumerate(x): mask[e_id, src_len:] = 1mask = mask.to(device = 'cpu')原创 2022-01-22 13:22:27 · 1871 阅读 · 0 评论