pytorch
xjh0929
认真生活
展开
-
深度学习常见报错
1.BrokenPipeError: [Errno 32] Broken pipe将DataLoader函数里面的num_workers参数减小或者修改为0原创 2021-12-27 10:09:12 · 814 阅读 · 0 评论 -
tensor常用函数归纳
1*.contiguous()* 这个函数会强制复制一份tensor,不会对原始的tensor做出改变。2.matmul 函数返回两个数组的矩阵乘积3.detach就是截断反向传播的梯度流。原创 2021-11-04 15:44:42 · 381 阅读 · 0 评论 -
交叉熵损失函数权重计算
交叉熵损失函数torch.nn.CrossEntropyLoss(weight=None, size_average=None, ignore_index=-100, reduce=None, reduction='elementwise_mean')weight为每个类别的loss设置权重,常用于类别不均衡的问题,weight的数据类型必须是float的tensor,其个数要与分类的类别个数一致(其他参数不介绍)weight的计算方法经常使用median frequency balanci原创 2021-07-02 10:12:21 · 4296 阅读 · 2 评论 -
labelimg的使用教程
1.安装cmd进入控制台程序,然后输入pip install labelimg2.打开cmd进入控制台程序,然后输入labelimg,回车即可3.使用点击open dir选择图片的路径(例如自己创建的一个photo文件夹,并把原始的图片放在这里面),点击change save dir选择保存的路径(Annotations文件夹,这里是存放勾画结果的)。在view菜单栏中打开auto save mode格式,这个可以帮助我们自动的保存。然后点击creat\nRectBox进行勾画,勾画之后就可原创 2020-07-15 16:01:38 · 7743 阅读 · 0 评论