自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(3)
  • 收藏
  • 关注

转载 使用Pytorch构建网络时,损失函数需要迁移到GPU上吗?

再进一步,如CrossEntropyCriterion等pytorch自带的loss类(或称losslayer)虽然也是Module的子类,但是这些类的成员变量不含Parameter对象或Buffer对象,所以无需to或cuda处理。模型输入tensor是数据不用多说,Module对象使用.cuda是为了将模型内部的数据(即每个子叶Module的Parameter或Buffer成员变量)递归地传入gpu。...

2022-08-01 10:22:20 1060 1

原创 图像读取格式转换

图像读取格式tensor numpy PIL相互转换

2022-07-26 11:03:21 187 1

原创 KMP算法如何求next[ ]值,超级好记的方法

模式中第j个字符失配时:                           next[j]=0  (如果j=1);                      next[j]=1  (如果j=2);                       next[j]={前(j-1)个字符中:前缀和后缀最大的完全相同长度+1}   (如果j>=3);  next[j] 的含义是...

2018-09-20 11:30:53 2037

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除