Pytorch
ACE-Mayer
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
Pytorch中nn.Conv2d的dilation
dilation原文解释如下:controls the spacing between the kernel points; alsoknown as the à trous algorithm. It is harder to describe, but this link_has a nice visualization of what :attr:dilation does.原文中也说描述起来有些难度,就是卷积核的各元素间隔开,默认间隔大小为1。如下图所示:...原创 2021-04-06 22:31:41 · 1802 阅读 · 0 评论 -
修改pytorch和Keras预训练模型路径
目录1、Pytorch预训练模型路径修改2、Keras修改预训练模型位置1、Pytorch预训练模型路径修改Pytorch安装目录下有一个hub.py,改文件指定了预训练模型的加载位置。该文件存在于xxx\site-packages\torch,例如我的存在于“C:\ProgramData\Miniconda3\Lib\site-packages\torch”。打开hub.py文件,找到load_state_dict_from_url函数,其中第二个参数model_dir用于指定权重文件路径:mod原创 2021-03-23 16:16:46 · 1349 阅读 · 0 评论 -
Pytorch中loss计算解析
下面是一段Pytorch分类案例代码:running_loss = 0.0running_corrects = 0# Iterate over data.for inputs, labels in dataloaders[phase]: inputs = inputs.to(device) labels = labels.to(device) # zero the parameter gradients optimizer.zero_grad() # forward #原创 2021-03-28 16:52:41 · 947 阅读 · 0 评论
分享