pytorch代码
关注
分享
扫一扫
山遥路源
这个作者很懒，什么都没留下…
展开
-
python读取图片的不同方式及转换
pil读取的图像模式是RGB，张量形状为（H，W，C），此时需要将HWC转为CHW。保存图像，与读取图像相反的思路即可。原创 2025-02-04 16:32:54 · 161 阅读 · 0 评论 -
module.register_forward_hook这个函数的含义
此函数是 在前向传播中添加“钩子”，在前向传播过程中，自动调用钩子函数。此“钩子”函数可自定义。原创 2025-01-24 13:34:17 · 239 阅读 · 0 评论 -
model.eval()与requires_grad
用于将该模型设置为评估模式，在训练过程中，不会进行反向传播和参数更新。原创 2025-01-06 16:22:05 · 130 阅读 · 0 评论 -
模型相关的好用函数
官方文档解释：给定 被注册过的模型名称 和 指定的模型参数 ，返回一个实例化的模型name:**config:返回一个分类的类别数为100的resnet18模型。原创 2025-01-06 17:03:14 · 197 阅读 · 0 评论 -
lambda
会返回一个类似指针的东西，可赋给某变量【可将函数当作参数传递】原创 2025-01-06 17:29:28 · 116 阅读 · 0 评论