深度学习
dream6104
仰望星空,脚踏实地。
展开
-
pytorch笔记-实现一个图像分类模型
定义一个神经网络结构1.数据引入import torch from torch import nn from torch.utils.data import DataLoader from torchvision import datasets from torchvision.transforms import ToTensor2.训练集与测试集我们用到的数据集是FashionMNIST,是一个图像数据集,用它来进行分类任务。dataloader用来存放相应的训练数据以及对应的标签da原创 2022-04-26 11:13:52 · 2029 阅读 · 0 评论 -
BERT模型自定义词汇以及token相关
1.加载bert模型及分词from transformers import AutoModelForMaskedLM, AutoTokenizermodel = "bert-base-cased"tokenizer = AutoTokenizer.from_pretrained(model, use_fast=True)model = AutoModelForMaskedLM.from_pretrained(model)2.分词演示这里是对COVID hospitalization分词p原创 2021-08-17 22:38:15 · 2444 阅读 · 1 评论