- 博客(6)
- 资源 (3)
- 收藏
- 关注
RSS订阅原创 Pytorch报错：Assertionerror nn criterions don't compute the gradient w.r.t. targets
引发错误目前看来有两种情况： 1、应该是在损失函数中，模型输出为首个实参，ground_truth为第二个实参，写反了会出现这种情况loss = criterion(output,label) #正确loss = criterion(label,output) #错误2、第二种情况，ground_truth需要与模型的某些参数运算后得到模型需要的标签，然后引发错误：#错误写...
2018-08-25 15:47:09
1381
转载 Pytorch画模型图
pytorch中好像没有个api让我们直观的看到模型的样子。但是有网友提供了一段代码，可以把模型画出来，对我来说简直就是如有神助啊。话不多说，上代码吧。import torchfrom torch.autograd import Variableimport torch.nn as nnfrom graphviz import Digraphclass CNN(nn.Module)...
2018-08-21 15:56:33
4192
原创 关于Pytorch 0.3 nn.Module的子类，前向传播过程的问题
线上代码：class ft_net(nn.Module): def __init__(self, class_num ): super(ft_net, self).__init__() model_ft = models.resnet50(pretrained=True) # avg pooling to global poolin...
2018-08-20 10:33:02
887
原创 tqdm介绍及常用方法
Tqdm 是一个快速，可扩展的Python进度条，可以在 Python 长循环中添加一个进度提示信息，用户只需要封装任意的迭代器 tqdm(iterator)。 使用pip就可以安装。使用方法一: tqdmtqdm(list)方法可以传入任意一种list,比如数组from tqdm import tqdmfor i in tqdm(range(1000)): #do ...
2018-08-16 20:17:39
272168
23
原创 Keras预训练模型下载后保存路径
ResNet50等keras预训练权重文件的下载：https://github.com/fchollet/deep-learning-models/releases/本地存放目录： Linux下是放在“~/.keras/models/”中 Win下则放在Python的“settings/.keras/models/”中 在anaconda on win中是：D:\Anacond...
2018-08-11 20:21:12
8243
3
Python计算机视觉（全书）
2017-09-18
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人