首页 文章 精选 留言 我的

精选列表

搜索[K2.6模型],共10000篇文章
优秀的个人博客,低调大师

一个 Transformer 训练生成式模型的例子

最近在看chatGPT,想着chatGPT 是怎么训练出来的,不涉及神经网络算法,可以使用Transformer玩一下 import torch import torch.nn as nn import torch.nn.functional as F from torch.utils.data import Dataset, DataLoader # 构造词表 vocab = ["<PAD>", "<BOS>", "<EOS>", "明天", "天气", "很", "好"] word2idx = {w: i for i, w in enumerate(vocab)} idx2word = {i: w for w, i in word2idx.items()} # 示例训练数据:输入和标签都偏移一位 # 输入: <BOS> 明天 天气 很 # 输出: 明天 天气 很 好 inputs = torch.tensor([ [word2idx["<BOS>"], word2idx["明天"], word2idx["天气"], word2idx["很"]] ]) labels = torch.tensor([ [word2idx["明天"], word2idx["天气"], word2idx["很"], word2idx["好"]] ]) class TinyTransformerModel(nn.Module): def __init__(self, vocab_size, d_model=32, nhead=2, num_layers=2): super().__init__() self.embedding = nn.Embedding(vocab_size, d_model) self.pos_encoding = nn.Parameter(torch.randn(1, 100, d_model)) # 最多100个词 encoder_layer = nn.TransformerEncoderLayer(d_model, nhead) self.transformer = nn.TransformerEncoder(encoder_layer, num_layers) self.fc = nn.Linear(d_model, vocab_size) def forward(self, x): x = self.embedding(x) + self.pos_encoding[:, :x.size(1)] x = self.transformer(x) return self.fc(x) model = TinyTransformerModel(vocab_size=len(vocab)) optimizer = torch.optim.Adam(model.parameters(), lr=0.01) loss_fn = nn.CrossEntropyLoss() for epoch in range(50): model.train() out = model(inputs) # [batch_size, seq_len, vocab_size] loss = loss_fn(out.view(-1, len(vocab)), labels.view(-1)) optimizer.zero_grad() loss.backward() optimizer.step() if epoch % 20 == 0: print(f"Epoch {epoch}, loss: {loss.item():.4f}") def generate(model, start_tokens, max_len=5): model.eval() input_ids = torch.tensor([start_tokens]) for _ in range(max_len): with torch.no_grad(): logits = model(input_ids) next_token = logits[0, -1].argmax().item() input_ids = torch.cat([input_ids, torch.tensor([[next_token]])], dim=1) if next_token == word2idx["<EOS>"] or len(input_ids[0]) > max_len: break return [idx2word[i] for i in input_ids[0].tolist()] # 测试 generated = generate(model, [word2idx["<BOS>"]]) print("生成结果:", " ".join(generated))

资源下载

更多资源
Mario

Mario

马里奥是站在游戏界顶峰的超人气多面角色。马里奥靠吃蘑菇成长,特征是大鼻子、头戴帽子、身穿背带裤,还留着胡子。与他的双胞胎兄弟路易基一起,长年担任任天堂的招牌角色。

Nacos

Nacos

Nacos /nɑ:kəʊs/ 是 Dynamic Naming and Configuration Service 的首字母简称,一个易于构建 AI Agent 应用的动态服务发现、配置管理和AI智能体管理平台。Nacos 致力于帮助您发现、配置和管理微服务及AI智能体应用。Nacos 提供了一组简单易用的特性集,帮助您快速实现动态服务发现、服务配置、服务元数据、流量管理。Nacos 帮助您更敏捷和容易地构建、交付和管理微服务平台。

Spring

Spring

Spring框架(Spring Framework)是由Rod Johnson于2002年提出的开源Java企业级应用框架,旨在通过使用JavaBean替代传统EJB实现方式降低企业级编程开发的复杂性。该框架基于简单性、可测试性和松耦合性设计理念,提供核心容器、应用上下文、数据访问集成等模块,支持整合Hibernate、Struts等第三方框架,其适用范围不仅限于服务器端开发,绝大多数Java应用均可从中受益。

Rocky Linux

Rocky Linux

Rocky Linux(中文名:洛基)是由Gregory Kurtzer于2020年12月发起的企业级Linux发行版,作为CentOS稳定版停止维护后与RHEL(Red Hat Enterprise Linux)完全兼容的开源替代方案,由社区拥有并管理,支持x86_64、aarch64等架构。其通过重新编译RHEL源代码提供长期稳定性,采用模块化包装和SELinux安全架构,默认包含GNOME桌面环境及XFS文件系统,支持十年生命周期更新。

用户登录
用户注册