![](https://img-blog.csdnimg.cn/20190927151053287.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
Pytorch
文章平均质量分 64
深度学习框架pytorch使用相关
lishanlu136
关注机器视觉,深度学习图像分类,人脸识别,目标检测/跟踪等方面,我的github账号:https://github.com/lishanlu136,欢迎交流。
展开
-
pytorch的model.parameters(),model.children(),model.modules(),model.named_parameters(),...
关于model.parameters()、model.children()、model.modules()、model.named_parameters()、model.named_children()、model.named_modules()的总结。原创 2023-09-22 17:16:05 · 227 阅读 · 0 评论 -
pytorch冻结参数训练的坑
pytorch冻结层遇到的坑,冻结Batchnorm层原创 2023-09-22 15:33:38 · 1060 阅读 · 0 评论 -
yolox训练自己的数据
此文是我从yolov5替换到yolox训练的过程,前提是我们有图片和标注文件,而且都是yolov5的txt格式的原创 2021-12-23 17:06:22 · 5132 阅读 · 8 评论 -
pytorch优化器传入两个模型的参数/保存和加载两个模型的参数
在人脸识别任务中,当我定义模型backbone后,用到Arcface_loss,但这个Arcface_loss也是用nn.Module模块写的,所以实例化出来也是一个网络,而且原论文中,Arcface_loss还是以backbone权重参数10倍的权重衰减方式更新,需要单独以不同的方式训练,且网络中bn层也是不需要权重衰减的。由于这些原因,我们就需要把网络参数分开。分别传入两个模型的参数如果我们只有一个模型model,我们一般就是这样初始化优化器的:optimizer = torch.optim..原创 2021-11-12 11:41:15 · 6425 阅读 · 0 评论 -
Linux终端运行python命令卡死,torch.utils.cpp_extension.load 无响应?
大体情况是这样的,我在本地用xshell连接到远程服务器,用xshell的界面打开服务器终端,通过Anaconda虚拟环境执行工程服务文件,执行命令是这样的:CUDA_VISIBLE_DEVICES=1 python service.py,当我在终端敲完这条命令,点击enter键,终端没任何反应(排除掉ctrl+s锁死终端界面的情况,因为我还能输入命令,按enter键后,光标能跳到下一行)附上终端的快捷键命令:ctrl+s 暂停屏幕输出ctrl+q 恢复屏幕输出ctrl+l 清屏,..原创 2021-11-04 16:17:50 · 3138 阅读 · 6 评论 -
Pytorch做数据扩充(torchvision.transforms的使用)
在深度学习训练过程中,经常会用到数据扩充,以下是用pytorch中torchvision.transforms模块对图片做增强的例子,在此做过实验记录。Pytorch做数据扩充transforms.ToTensor()原创 2021-01-14 21:27:18 · 2154 阅读 · 0 评论