分类：预训练

2020
11-19

PyTorch预训练Bert模型的示例

本文介绍以下内容：1.使用transformers框架做预训练的bert-base模型；2.开发平台使用Google的Colab平台，白嫖GPU加速；3.使用datasets模块下载IMDB影评数据作为训练数据。transformers模块简介transformers框架为Huggingface开源的深度学习框架，支持几乎所有的Transformer架构的预训练模型。使用非常的方便，本文基于此框架，尝试一下预训练模型的使用，简单易用。本来打算预训练bert-large模型，发现colab上GPU显存不够用，只... 继续阅读 >

2020
10-08

Keras预训练的ImageNet模型实现分类操作

本文主要介绍通过预训练的ImageNet模型实现图像分类，主要使用到的网络结构有：VGG16、InceptionV3、ResNet50、MobileNet。代码：importkerasimportnumpyasnpfromkeras.applicationsimportvgg16,inception_v3,resnet50,mobilenet#加载模型vgg_model=vgg16.VGG16(weights='imagenet')inception_model=inception_v3.InceptionV3(weights='imagenet')resnet_model=resnet50.ResNet50(weights='imagenet')mobilenet_... 继续阅读 >

2020
10-08

解决Keras自带数据集与预训练model下载太慢问题

keras的数据集源码下载地址太慢。尝试过修改源码中的下载地址，直接报错。从源码或者网络资源下好数据集，下载好以后放到目录 ~/.keras/datasets/   下面。其中：cifar10需要改文件名为cifar-10-batches-py.tar.gz，cifar100改为cifar-100-python.tar.gz,mnist改为mnist.npz预训练models放到~/.keras/models/路径下面即可。补充知识：Keras下载的数据集以及预训练模型保存在哪里Keras下载的数据集在... 继续阅读 >