【pytorch】预训练模型的使用;修改模型结构;自定义模型;给模型添加层
pytorch自带有一些高级的复杂模型,我们可以通过 torchvision.models
调用,例如torchvision.models.densenet169(pretrained=True)
就调用了densenet169的预训练模型。
目录
先来看看这个模块涵盖了哪些模型:
见https://pytorch.org/vision/stable/models.html获取详细信息。
注意
基本所有预训练模型都是基于ImageNet数据集训练的,得先从从网站(https://s3.amazonaws.com/deep-learning-models/imag