- 博客(5)
- 收藏
- 关注
原创 DataWhale组队学习-PyTorch 可视化学习
可视化网络结构使用torchinfo工具包来可视化网络结构。使用print函数打印模型基础信息以ResNet18为例import torchvision.models as modelsmodel = models.resnet18()print(model)这时候得到的模型信息输出结果只能得出基础构件的信息,既不能显示出每一层的shape,也不能显示对应参数量的大小。具体结果如下:ResNet( (conv1): Conv2d(3, 64, kernel_size=(7, 7), s
2022-03-22 22:12:52
581
原创 Python3 命名空间和作用域
命名空间(Namespace)一个例子有三种类别:内置名称(build-in names): Python 语言内置的名称全局名称(global names): 模块中定义的名称,记录了模块的变量,包括函数、类、其它导入的模块、模块级的变量和常量。局部名称(local names): 函数中定义的名称,记录了函数的变量,包括函数的参数和局部定义的变量。(类中定义的也是)命名空间的查找顺序:局部的命名空间去 -> 全局命名空间 -> 内置命名空间若找不到则报错:Name
2022-03-19 21:39:17
725
原创 Python函数装饰器
0. 函数的使用方法除了基本的函数定义外,还可以在函数中定义函数,或者返回一个函数,或者将函数作为参数传给另一个函数。1. 装饰器的功能装饰器用来封装一个函数,并且用这样或者那样的方式来修改它的行为。例如def a_new_decorator(a_func): def wrapTheFunction(): print("I am doing some boring work before executing a_func()") a_func()
2022-03-19 21:34:17
487
原创 Datawhale组队学习——Pytorch进阶训练技巧
自定义损失函数以函数的方式定义def my_loss(output, target): loss = torch.mean((output - target)**2) return loss以类方式定义更多的时候以类的方式定义,观察Pytorch自带的损失函数,部分损失函数直接继承自_Loss类,部分则先继承自_WeightedLoss类,而_WeightedLoss又继承自_Loss类。_Loss类则最终继承自nn.Module。_Loss类的定义如下:class _Loss
2022-03-19 21:26:06
360
原创 PyTorch模型定义
PyTorch模型定义应包括两个主要部分:各个部分的初始化(_init_); 数据流向定义(forward)。基于nn.Module,可以通过Sequential,ModuleList和ModuleDict三种方式定义PyTorch模型。方式1-Sequentialnn.Sequential()接收一系列子模块或者一个子模块的有序字典(OrderedDict) 作为参数来逐一添加 Module 的实例。使用Sequential定义的模型可以不重写forward适用场景:...
2022-03-16 23:35:24
901
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人