简介本资源面向希望入门多模态大模型开发的开发者与研究者围绕CLIP模型构建了一套完整的「看图说话」图像描述生成实战项目帮助读者理解视觉与文本跨模态对齐的核心思路并掌握从数据预处理到模型训练、推理部署的全流程。压缩包共52个文件约4.87MB包含7个Python脚本、4个Shell训练与预测脚本、3个JSON配置、8个TXT数据说明及26张JPG示例图片另附PNG与Markdown文档覆盖数据集构建、模型定义、训练与预测等关键环节。项目基于Flickr图文数据提供GPT-2微调与非微调两套实验路径并配有损失曲线、模型结构等可视化图便于对照理解训练效果。目前已有281人学习下载适合作为多模态大模型应用开发的入门实战参考。1. 从一张图到一句话CLIP GPT2 看图说话项目能跑出什么你手里有一张图想让机器自己说出一句像样的描述——不是打标签那种「狗、草地、飞盘」而是「一只狗在草地上接飞盘」这种带主谓宾的句子。这个项目就是干这个的用 CLIP 做图像编码接一个 GPT2 解码器把视觉特征翻译成自然语言。源码包里带了process_caption.py、train.py、predict.py三个核心脚本还有 Flickr 数据集的 caption 文件、预训练模型目录、四张训练曲线图loss、transformer、mlp、overview以及四个 shell 脚本分别对应「不微调」「微调 GPT2」「微调 MLP」三种训练/推理组合。适合谁已经跑通过 PyTorch 基础训练、想搞明白多模态特征怎么对齐、怎么把 CLIP 的 embedding 喂进语言模型的从业者。新手也能跟但得先把环境搭对不然第一步就卡在 CLIP 权重下载上。2. 拆开看结构CLIP 编码器 映射层 GPT2 解码器怎么串2.1 为什么选 CLIP 做视觉端而不是 ResNetCLIP 的核心价值在于它用 4 亿对图文数据做了对比学习图像编码器输出的 512 维向量本身就带着「语义」——它和文本编码器在同一空间里对齐过。这意味着你拿 CLIP 的 image embedding 去接一个语言模型比拿 ResNet 的 pool5 特征去接收敛快得多。ResNet 的特征是分类任务训出来的它关心的是「这是不是狗」不关心「狗在干什么」。CLIP 的特征关心的是「这张图和哪段文字更像」天然适合生成任务。项目里model.py定义了一个CaptionModel结构是# model.py 核心结构示意 import torch import torch.nn as nn from transformers import GPT2LMHeadModel class CaptionModel(nn.Module): def __init__(self, clip_dim512, gpt2_dim768, prefix_len10): super().__init__() # 映射层把 CLIP 的 512 维投影到 GPT2 的 768 维 self.mlp nn.Sequential( nn.Linear(clip_dim, gpt2_dim), nn.ReLU(), nn.Linear(gpt2_dim, gpt2_dim) ) # prefix_len 个可学习的 prefix token替代原始图像 token self.prefix nn.Parameter(torch.randn(1, prefix_len, gpt2_dim)) self.gpt2 GPT2LMHeadModel.from_pretrained(gpt2) def forward(self, clip_feat, input_ids, attention_mask): # clip_feat: [B, 512] - [B, 1, 768] mapped self.mlp(clip_feat).unsqueeze(1) # 拼接 prefix 和映射后的图像特征 prefix self.prefix.expand(clip_feat.size(0), -1, -1) visual_token torch.cat([mapped, prefix], dim1) # [B, 11, 768] # 把 visual_token 拼到文本 embedding 前面 text_emb self.gpt2.transformer.wte(input_ids) inputs_embeds torch.cat([visual_token, text_emb], dim1) outputs self.gpt2(inputs_embedsinputs_embeds) return outputs.logits逻辑说明CLIP 输出的是全局池化后的 512 维向量直接送进 GPT2 的 embedding 层维度对不上所以加一个两层 MLP 做投影。prefix_len10是超参意思是除了映射后的那一个视觉 token再补 10 个可学习的 prefix token让模型有更多容量去适配视觉信息。参数怎么改如果你的显存够prefix_len可以加到 20但超过 20 之后增益很小反而容易过拟合。mlp中间层的 ReLU 可以换成 GELU实测在 Flickr 数据上收敛更稳。2.2 三种训练模式不微调、微调 MLP、微调 GPT2项目给了四个 shell 脚本对应三种模式脚本训练部分推理脚本显存占用适用场景train_no_finetune_gpt2.sh只训 MLP 映射层predict_no_finerune_gpt2.sh~6GB快速验证、数据量小train_finetune_gpt2.shMLP GPT2 全部微调predict_finerune_gpt2.sh~14GB追求生成质量train_no_finetune_gpt2.sh 手动改只训 prefix同上~8GB折中方案不微调 GPT2 的意思是GPT2 的 1.24 亿参数全部冻结只训练 MLP 和 prefix。这样做的好处是训练快在 Flickr 的 3 万张图上单卡 2080Ti 大概 40 分钟能跑完 10 个 epoch。坏处是生成句子的多样性差容易输出「a dog is running in the grass」这种模板句。微调 GPT2 之后模型会学会更丰富的表达但需要小心过拟合——Flickr 数据集的 caption 平均长度只有 12 个词微调太狠会让模型只会说短句。我一般会先跑不微调版本确认数据管道和 loss 下降正常再切到微调版本。train.py里有个--finetune_gpt2的 flag默认是 False改成 True 就切模式。2.3 数据预处理process_caption.py和dataset.py的分工process_caption.py负责把flickr_caption.txt里的原始 caption 做清洗去掉标点、转小写、加|startoftext|和|endoftext|标记。dataset.py里的FlickrDataset类负责在__getitem__里同时返回图像张量和 tokenized 后的 caption。# dataset.py 关键片段 from torch.utils.data import Dataset from PIL import Image import torch class FlickrDataset(Dataset): def __init__(self, caption_file, image_dir, tokenizer, transform, max_len32): self.data [] with open(caption_file, r, encodingutf-8) as f: for line in f: img_name, caption line.strip().split(\t) self.data.append((img_name, caption)) self.image_dir image_dir self.tokenizer tokenizer self.transform transform self.max_len max_len def __getitem__(self, idx): img_name, caption self.data[idx] image Image.open(f{self.image_dir}/{img_name}).convert(RGB) image self.transform(image) # CLIP 的预处理resize 224 normalize tokens self.tokenizer( caption, paddingmax_length, max_lengthself.max_len, truncationTrue, return_tensorspt ) return image, tokens[input_ids].squeeze(0), tokens[attention_mask].squeeze(0)参数说明max_len32是 caption 的最大 token 数Flickr 的 caption 很少超过 20 个 token设 32 是留余量。transform必须用 CLIP 自己的预处理不能用 ImageNet 的 mean/std否则 CLIP 编码器输出的特征分布会偏。常见错误是直接拿torchvision.transforms.ToTensor()就送进去结果 loss 不降——因为 CLIP 期望的输入是(x - mean) / stdmean 和 std 是 CLIP 训练时统计的。3. 跑通训练和推理从train.py到predict.py的完整链路3.1 环境准备和依赖安装requirements.txt里列了核心依赖但没锁版本。我建议手动锁一下避免 transformers 版本更新后 API 变了# 建议的安装顺序 pip install torch1.13.1cu117 torchvision0.14.1cu117 -f https://download.pytorch.org/whl/torch_stable.html pip install transformers4.28.0 pip install ftfy regex tqdm pip install githttps://github.com/openai/CLIP.git注意CLIP 的安装必须从 GitHub 装PyPI 上的clip包是另一个东西。ftfy是 CLIP 的依赖用来处理文本里的特殊字符。如果你用 conda先conda install pytorch torchvision cudatoolkit11.7 -c pytorch再 pip 装剩下的。预训练模型目录pretrain_models里应该放两个东西CLIP 的权重ViT-B-32.pt和 GPT2 的权重从 HuggingFace 下载的gpt2文件夹。CLIP 权重下载地址在 OpenAI 的 GitHub 仓库里有GPT2 用GPT2LMHeadModel.from_pretrained(gpt2)会自动下载到~/.cache/huggingface。如果服务器没外网提前把这两个权重下好放进去。3.2 训练脚本参数逐项拆解train.py的 argparse 部分定义了所有可调参数我挑几个关键的讲python train.py \ --caption_file datasets/flickr_caption.txt \ --image_dir datasets/flickr30k/images \ --pretrain_model pretrain_models/gpt2 \ --clip_model ViT-B-32 \ --batch_size 64 \ --lr 1e-4 \ --epochs 10 \ --prefix_len 10 \ --finetune_gpt2 False \ --save_path output/bert_no_finetune_gpt2--batch_size 64在 2080Ti 上跑不微调模式刚好微调模式要降到 16 或 32。--lr 1e-4是 AdamW 的初始学习率微调 GPT2 时建议降到 5e-5否则容易把预训练学到的语言知识冲掉。--prefix_len 10前面说过10 是性价比最高的值。--save_path对应输出目录项目里已经建好了bert_no_finetune_gpt2和mlp_finetune_gpt2两个文件夹分别对应两种模式。训练过程中会打印 loss正常情况下前 200 步 loss 从 8 左右降到 4 左右然后缓慢下降到 2.5 附近。如果 loss 卡在 7 不动检查三件事图像预处理是不是用了 CLIP 的 transform、CLIP 权重是不是加载成功、caption 里的特殊 token 是不是和 GPT2 的 tokenizer 对上了。3.3 推理predict.py怎么把图像变成句子推理脚本的核心是自回归生成先拿 CLIP 编码图像映射后拼上 prefix然后让 GPT2 一个 token 一个 token 地往外吐。# predict.py 核心逻辑 import torch from PIL import Image import clip def generate_caption(model, image_path, tokenizer, clip_model, clip_preprocess, device, max_len30): image Image.open(image_path).convert(RGB) image_input clip_preprocess(image).unsqueeze(0).to(device) with torch.no_grad(): clip_feat clip_model.encode_image(image_input).float() # 从 |startoftext| 开始生成 input_ids torch.tensor([[tokenizer.bos_token_id]]).to(device) for _ in range(max_len): logits model(clip_feat, input_ids, attention_maskNone) next_token logits[:, -1, :].argmax(dim-1, keepdimTrue) input_ids torch.cat([input_ids, next_token], dim1) if next_token.item() tokenizer.eos_token_id: break caption tokenizer.decode(input_ids[0], skip_special_tokensTrue) return caption逻辑说明clip_model.encode_image返回的是归一化后的特征.float()是因为 CLIP 在某些精度下会输出 half。argmax是贪心解码生成结果确定但多样性差。想换 beam search 的话把argmax换成torch.topk加长度惩罚但代码量会增加。max_len30是硬截断防止模型陷入重复循环。常见翻车点如果生成的句子全是「a a a a」说明 prefix 没训好或者学习率太大把 embedding 打散了。4. 避坑与排查训练不收敛、生成重复、显存爆炸的常见原因4.1 loss 不下降一直卡在 7 以上现象训练跑了 500 步loss 从 8 降到 7.8 就不动了。原因CLIP 的图像预处理没用对。很多人直接拿transforms.ToTensor()就送进 CLIP但 CLIP 期望的输入是(pixel - mean) / stdmean 和 std 是[0.48145466, 0.4578275, 0.40821073]和[0.26862954, 0.26130258, 0.27577711]。解决用clip_preprocess或者手动写 Normalize。4.2 生成的 caption 重复同一个词现象推理输出「a dog dog dog dog」。原因prefix token 的初始化方差太大或者学习率过高导致 prefix 在训练早期就发散。解决把prefix的初始化改成torch.randn(...) * 0.02学习率降到 5e-5并在 loss 里加一点重复惩罚repetition penalty。4.3 微调 GPT2 时显存爆炸现象--finetune_gpt2 True跑起来直接 OOMbatch_size 降到 8 还是炸。原因GPT2 的 1.24 亿参数全部参与梯度更新加上 CLIP 的前向传播显存占用是推理时的 3 倍以上。解决用 gradient checkpointing在model.py里给 GPT2 加self.gpt2.gradient_checkpointing_enable()显存能降 40%代价是训练速度慢 20%。4.4 推理时生成的句子和图像无关现象给一张狗的图生成「a cat is sitting on the sofa」。原因CLIP 特征和 GPT2 的 embedding 空间没对齐MLP 映射层没训够。解决先冻结 GPT2 单独训 MLP 5 个 epoch等 loss 降到 3 以下再解冻 GPT2 微调。项目里的train_no_finetune_gpt2.sh就是干这个的。4.5 加载预训练权重时报 key 不匹配现象load_state_dict报Missing key(s) in state_dict: gpt2.transformer.wte.weight。原因保存模型时用了torch.save(model.state_dict())但加载时模型结构变了比如 prefix_len 改了。解决确保加载时的prefix_len和保存时一致或者用strictFalse忽略不匹配的 key但这样会丢参数。5. 进阶技巧用 prefix 调参和 beam search 把生成质量再拉一档5.1 prefix_len 和学习率的联合调参prefix_len不是越大越好。我做过一组对比实验在 Flickr 数据集上固定其他参数只改prefix_lenprefix_lenBLEU-4训练时间epoch显存增量50.2112min0.3GB100.2614min0.5GB200.2718min0.9GB500.2528min2.1GB10 到 20 之间 BLEU 只涨了 0.01但训练时间多了 30%。所以 10 是甜点值。学习率方面不微调 GPT2 时用 1e-4微调时用 5e-5如果 loss 震荡就再降一半。5.2 把贪心解码换成 beam searchpredict.py默认是贪心生成结果死板。改成 beam search 只需要改几行# beam search 版本 def beam_search_caption(model, clip_feat, tokenizer, device, beam_width3, max_len30): beams [([tokenizer.bos_token_id], 0.0)] # (token_list, score) for _ in range(max_len): new_beams [] for tokens, score in beams: input_ids torch.tensor([tokens]).to(device) logits model(clip_feat, input_ids, attention_maskNone) log_probs torch.log_softmax(logits[:, -1, :], dim-1) topk_log_probs, topk_ids log_probs.topk(beam_width) for i in range(beam_width): new_tokens tokens [topk_ids[0][i].item()] new_score score topk_log_probs[0][i].item() new_beams.append((new_tokens, new_score)) # 按长度归一化后排序保留 beam_width 个 beams sorted(new_beams, keylambda x: x[1] / len(x[0]), reverseTrue)[:beam_width] if all(tokens[-1] tokenizer.eos_token_id for tokens, _ in beams): break best_tokens beams[0][0] return tokenizer.decode(best_tokens, skip_special_tokensTrue)逻辑说明beam search 维护beam_width条候选路径每步扩展后按长度归一化的 log 概率排序。beam_width3在 Flickr 上比贪心 BLEU 高 0.03 左右但推理时间翻倍。如果只是 demo贪心够用如果要写论文对比beam search 是标配。5.3 验证模型有没有真正学会「看图」一个简单的 sanity check拿同一张图分别用「不微调」和「微调」两个模型生成 caption看后者是不是更具体。比如不微调可能输出「a dog in the grass」微调后输出「a brown dog running through the green grass」。如果两个模型输出一模一样说明微调没生效检查--finetune_gpt2是不是真的传进去了。还有一个更狠的验证把图像换成纯噪声看模型是不是还在生成「a dog」。如果噪声图也输出「a dog」说明模型根本没在看图只是在靠语言模型的先验瞎编。这时候要回去检查 CLIP 特征是不是真的拼进了 GPT2 的输入。从那以后我每次跑完训练都会先拿三张图做 sanity check一张训练集里的、一张训练集外的、一张纯噪声。三张图的输出如果符合预期才继续往下调参。希望帮到你。本文还有配套的精品资源点击获取
简介本资源面向希望入门多模态大模型开发的开发者与研究者围绕CLIP模型构建了一套完整的「看图说话」图像描述生成实战项目帮助读者理解视觉与文本跨模态对齐的核心思路并掌握从数据预处理到模型训练、推理部署的全流程。压缩包共52个文件约4.87MB包含7个Python脚本、4个Shell训练与预测脚本、3个JSON配置、8个TXT数据说明及26张JPG示例图片另附PNG与Markdown文档覆盖数据集构建、模型定义、训练与预测等关键环节。项目基于Flickr图文数据提供GPT-2微调与非微调两套实验路径并配有损失曲线、模型结构等可视化图便于对照理解训练效果。目前已有281人学习下载适合作为多模态大模型应用开发的入门实战参考。1. 从一张图到一句话CLIP GPT2 看图说话项目能跑出什么你手里有一张图想让机器自己说出一句像样的描述——不是打标签那种「狗、草地、飞盘」而是「一只狗在草地上接飞盘」这种带主谓宾的句子。这个项目就是干这个的用 CLIP 做图像编码接一个 GPT2 解码器把视觉特征翻译成自然语言。源码包里带了process_caption.py、train.py、predict.py三个核心脚本还有 Flickr 数据集的 caption 文件、预训练模型目录、四张训练曲线图loss、transformer、mlp、overview以及四个 shell 脚本分别对应「不微调」「微调 GPT2」「微调 MLP」三种训练/推理组合。适合谁已经跑通过 PyTorch 基础训练、想搞明白多模态特征怎么对齐、怎么把 CLIP 的 embedding 喂进语言模型的从业者。新手也能跟但得先把环境搭对不然第一步就卡在 CLIP 权重下载上。2. 拆开看结构CLIP 编码器 映射层 GPT2 解码器怎么串2.1 为什么选 CLIP 做视觉端而不是 ResNetCLIP 的核心价值在于它用 4 亿对图文数据做了对比学习图像编码器输出的 512 维向量本身就带着「语义」——它和文本编码器在同一空间里对齐过。这意味着你拿 CLIP 的 image embedding 去接一个语言模型比拿 ResNet 的 pool5 特征去接收敛快得多。ResNet 的特征是分类任务训出来的它关心的是「这是不是狗」不关心「狗在干什么」。CLIP 的特征关心的是「这张图和哪段文字更像」天然适合生成任务。项目里model.py定义了一个CaptionModel结构是# model.py 核心结构示意 import torch import torch.nn as nn from transformers import GPT2LMHeadModel class CaptionModel(nn.Module): def __init__(self, clip_dim512, gpt2_dim768, prefix_len10): super().__init__() # 映射层把 CLIP 的 512 维投影到 GPT2 的 768 维 self.mlp nn.Sequential( nn.Linear(clip_dim, gpt2_dim), nn.ReLU(), nn.Linear(gpt2_dim, gpt2_dim) ) # prefix_len 个可学习的 prefix token替代原始图像 token self.prefix nn.Parameter(torch.randn(1, prefix_len, gpt2_dim)) self.gpt2 GPT2LMHeadModel.from_pretrained(gpt2) def forward(self, clip_feat, input_ids, attention_mask): # clip_feat: [B, 512] - [B, 1, 768] mapped self.mlp(clip_feat).unsqueeze(1) # 拼接 prefix 和映射后的图像特征 prefix self.prefix.expand(clip_feat.size(0), -1, -1) visual_token torch.cat([mapped, prefix], dim1) # [B, 11, 768] # 把 visual_token 拼到文本 embedding 前面 text_emb self.gpt2.transformer.wte(input_ids) inputs_embeds torch.cat([visual_token, text_emb], dim1) outputs self.gpt2(inputs_embedsinputs_embeds) return outputs.logits逻辑说明CLIP 输出的是全局池化后的 512 维向量直接送进 GPT2 的 embedding 层维度对不上所以加一个两层 MLP 做投影。prefix_len10是超参意思是除了映射后的那一个视觉 token再补 10 个可学习的 prefix token让模型有更多容量去适配视觉信息。参数怎么改如果你的显存够prefix_len可以加到 20但超过 20 之后增益很小反而容易过拟合。mlp中间层的 ReLU 可以换成 GELU实测在 Flickr 数据上收敛更稳。2.2 三种训练模式不微调、微调 MLP、微调 GPT2项目给了四个 shell 脚本对应三种模式脚本训练部分推理脚本显存占用适用场景train_no_finetune_gpt2.sh只训 MLP 映射层predict_no_finerune_gpt2.sh~6GB快速验证、数据量小train_finetune_gpt2.shMLP GPT2 全部微调predict_finerune_gpt2.sh~14GB追求生成质量train_no_finetune_gpt2.sh 手动改只训 prefix同上~8GB折中方案不微调 GPT2 的意思是GPT2 的 1.24 亿参数全部冻结只训练 MLP 和 prefix。这样做的好处是训练快在 Flickr 的 3 万张图上单卡 2080Ti 大概 40 分钟能跑完 10 个 epoch。坏处是生成句子的多样性差容易输出「a dog is running in the grass」这种模板句。微调 GPT2 之后模型会学会更丰富的表达但需要小心过拟合——Flickr 数据集的 caption 平均长度只有 12 个词微调太狠会让模型只会说短句。我一般会先跑不微调版本确认数据管道和 loss 下降正常再切到微调版本。train.py里有个--finetune_gpt2的 flag默认是 False改成 True 就切模式。2.3 数据预处理process_caption.py和dataset.py的分工process_caption.py负责把flickr_caption.txt里的原始 caption 做清洗去掉标点、转小写、加|startoftext|和|endoftext|标记。dataset.py里的FlickrDataset类负责在__getitem__里同时返回图像张量和 tokenized 后的 caption。# dataset.py 关键片段 from torch.utils.data import Dataset from PIL import Image import torch class FlickrDataset(Dataset): def __init__(self, caption_file, image_dir, tokenizer, transform, max_len32): self.data [] with open(caption_file, r, encodingutf-8) as f: for line in f: img_name, caption line.strip().split(\t) self.data.append((img_name, caption)) self.image_dir image_dir self.tokenizer tokenizer self.transform transform self.max_len max_len def __getitem__(self, idx): img_name, caption self.data[idx] image Image.open(f{self.image_dir}/{img_name}).convert(RGB) image self.transform(image) # CLIP 的预处理resize 224 normalize tokens self.tokenizer( caption, paddingmax_length, max_lengthself.max_len, truncationTrue, return_tensorspt ) return image, tokens[input_ids].squeeze(0), tokens[attention_mask].squeeze(0)参数说明max_len32是 caption 的最大 token 数Flickr 的 caption 很少超过 20 个 token设 32 是留余量。transform必须用 CLIP 自己的预处理不能用 ImageNet 的 mean/std否则 CLIP 编码器输出的特征分布会偏。常见错误是直接拿torchvision.transforms.ToTensor()就送进去结果 loss 不降——因为 CLIP 期望的输入是(x - mean) / stdmean 和 std 是 CLIP 训练时统计的。3. 跑通训练和推理从train.py到predict.py的完整链路3.1 环境准备和依赖安装requirements.txt里列了核心依赖但没锁版本。我建议手动锁一下避免 transformers 版本更新后 API 变了# 建议的安装顺序 pip install torch1.13.1cu117 torchvision0.14.1cu117 -f https://download.pytorch.org/whl/torch_stable.html pip install transformers4.28.0 pip install ftfy regex tqdm pip install githttps://github.com/openai/CLIP.git注意CLIP 的安装必须从 GitHub 装PyPI 上的clip包是另一个东西。ftfy是 CLIP 的依赖用来处理文本里的特殊字符。如果你用 conda先conda install pytorch torchvision cudatoolkit11.7 -c pytorch再 pip 装剩下的。预训练模型目录pretrain_models里应该放两个东西CLIP 的权重ViT-B-32.pt和 GPT2 的权重从 HuggingFace 下载的gpt2文件夹。CLIP 权重下载地址在 OpenAI 的 GitHub 仓库里有GPT2 用GPT2LMHeadModel.from_pretrained(gpt2)会自动下载到~/.cache/huggingface。如果服务器没外网提前把这两个权重下好放进去。3.2 训练脚本参数逐项拆解train.py的 argparse 部分定义了所有可调参数我挑几个关键的讲python train.py \ --caption_file datasets/flickr_caption.txt \ --image_dir datasets/flickr30k/images \ --pretrain_model pretrain_models/gpt2 \ --clip_model ViT-B-32 \ --batch_size 64 \ --lr 1e-4 \ --epochs 10 \ --prefix_len 10 \ --finetune_gpt2 False \ --save_path output/bert_no_finetune_gpt2--batch_size 64在 2080Ti 上跑不微调模式刚好微调模式要降到 16 或 32。--lr 1e-4是 AdamW 的初始学习率微调 GPT2 时建议降到 5e-5否则容易把预训练学到的语言知识冲掉。--prefix_len 10前面说过10 是性价比最高的值。--save_path对应输出目录项目里已经建好了bert_no_finetune_gpt2和mlp_finetune_gpt2两个文件夹分别对应两种模式。训练过程中会打印 loss正常情况下前 200 步 loss 从 8 左右降到 4 左右然后缓慢下降到 2.5 附近。如果 loss 卡在 7 不动检查三件事图像预处理是不是用了 CLIP 的 transform、CLIP 权重是不是加载成功、caption 里的特殊 token 是不是和 GPT2 的 tokenizer 对上了。3.3 推理predict.py怎么把图像变成句子推理脚本的核心是自回归生成先拿 CLIP 编码图像映射后拼上 prefix然后让 GPT2 一个 token 一个 token 地往外吐。# predict.py 核心逻辑 import torch from PIL import Image import clip def generate_caption(model, image_path, tokenizer, clip_model, clip_preprocess, device, max_len30): image Image.open(image_path).convert(RGB) image_input clip_preprocess(image).unsqueeze(0).to(device) with torch.no_grad(): clip_feat clip_model.encode_image(image_input).float() # 从 |startoftext| 开始生成 input_ids torch.tensor([[tokenizer.bos_token_id]]).to(device) for _ in range(max_len): logits model(clip_feat, input_ids, attention_maskNone) next_token logits[:, -1, :].argmax(dim-1, keepdimTrue) input_ids torch.cat([input_ids, next_token], dim1) if next_token.item() tokenizer.eos_token_id: break caption tokenizer.decode(input_ids[0], skip_special_tokensTrue) return caption逻辑说明clip_model.encode_image返回的是归一化后的特征.float()是因为 CLIP 在某些精度下会输出 half。argmax是贪心解码生成结果确定但多样性差。想换 beam search 的话把argmax换成torch.topk加长度惩罚但代码量会增加。max_len30是硬截断防止模型陷入重复循环。常见翻车点如果生成的句子全是「a a a a」说明 prefix 没训好或者学习率太大把 embedding 打散了。4. 避坑与排查训练不收敛、生成重复、显存爆炸的常见原因4.1 loss 不下降一直卡在 7 以上现象训练跑了 500 步loss 从 8 降到 7.8 就不动了。原因CLIP 的图像预处理没用对。很多人直接拿transforms.ToTensor()就送进 CLIP但 CLIP 期望的输入是(pixel - mean) / stdmean 和 std 是[0.48145466, 0.4578275, 0.40821073]和[0.26862954, 0.26130258, 0.27577711]。解决用clip_preprocess或者手动写 Normalize。4.2 生成的 caption 重复同一个词现象推理输出「a dog dog dog dog」。原因prefix token 的初始化方差太大或者学习率过高导致 prefix 在训练早期就发散。解决把prefix的初始化改成torch.randn(...) * 0.02学习率降到 5e-5并在 loss 里加一点重复惩罚repetition penalty。4.3 微调 GPT2 时显存爆炸现象--finetune_gpt2 True跑起来直接 OOMbatch_size 降到 8 还是炸。原因GPT2 的 1.24 亿参数全部参与梯度更新加上 CLIP 的前向传播显存占用是推理时的 3 倍以上。解决用 gradient checkpointing在model.py里给 GPT2 加self.gpt2.gradient_checkpointing_enable()显存能降 40%代价是训练速度慢 20%。4.4 推理时生成的句子和图像无关现象给一张狗的图生成「a cat is sitting on the sofa」。原因CLIP 特征和 GPT2 的 embedding 空间没对齐MLP 映射层没训够。解决先冻结 GPT2 单独训 MLP 5 个 epoch等 loss 降到 3 以下再解冻 GPT2 微调。项目里的train_no_finetune_gpt2.sh就是干这个的。4.5 加载预训练权重时报 key 不匹配现象load_state_dict报Missing key(s) in state_dict: gpt2.transformer.wte.weight。原因保存模型时用了torch.save(model.state_dict())但加载时模型结构变了比如 prefix_len 改了。解决确保加载时的prefix_len和保存时一致或者用strictFalse忽略不匹配的 key但这样会丢参数。5. 进阶技巧用 prefix 调参和 beam search 把生成质量再拉一档5.1 prefix_len 和学习率的联合调参prefix_len不是越大越好。我做过一组对比实验在 Flickr 数据集上固定其他参数只改prefix_lenprefix_lenBLEU-4训练时间epoch显存增量50.2112min0.3GB100.2614min0.5GB200.2718min0.9GB500.2528min2.1GB10 到 20 之间 BLEU 只涨了 0.01但训练时间多了 30%。所以 10 是甜点值。学习率方面不微调 GPT2 时用 1e-4微调时用 5e-5如果 loss 震荡就再降一半。5.2 把贪心解码换成 beam searchpredict.py默认是贪心生成结果死板。改成 beam search 只需要改几行# beam search 版本 def beam_search_caption(model, clip_feat, tokenizer, device, beam_width3, max_len30): beams [([tokenizer.bos_token_id], 0.0)] # (token_list, score) for _ in range(max_len): new_beams [] for tokens, score in beams: input_ids torch.tensor([tokens]).to(device) logits model(clip_feat, input_ids, attention_maskNone) log_probs torch.log_softmax(logits[:, -1, :], dim-1) topk_log_probs, topk_ids log_probs.topk(beam_width) for i in range(beam_width): new_tokens tokens [topk_ids[0][i].item()] new_score score topk_log_probs[0][i].item() new_beams.append((new_tokens, new_score)) # 按长度归一化后排序保留 beam_width 个 beams sorted(new_beams, keylambda x: x[1] / len(x[0]), reverseTrue)[:beam_width] if all(tokens[-1] tokenizer.eos_token_id for tokens, _ in beams): break best_tokens beams[0][0] return tokenizer.decode(best_tokens, skip_special_tokensTrue)逻辑说明beam search 维护beam_width条候选路径每步扩展后按长度归一化的 log 概率排序。beam_width3在 Flickr 上比贪心 BLEU 高 0.03 左右但推理时间翻倍。如果只是 demo贪心够用如果要写论文对比beam search 是标配。5.3 验证模型有没有真正学会「看图」一个简单的 sanity check拿同一张图分别用「不微调」和「微调」两个模型生成 caption看后者是不是更具体。比如不微调可能输出「a dog in the grass」微调后输出「a brown dog running through the green grass」。如果两个模型输出一模一样说明微调没生效检查--finetune_gpt2是不是真的传进去了。还有一个更狠的验证把图像换成纯噪声看模型是不是还在生成「a dog」。如果噪声图也输出「a dog」说明模型根本没在看图只是在靠语言模型的先验瞎编。这时候要回去检查 CLIP 特征是不是真的拼进了 GPT2 的输入。从那以后我每次跑完训练都会先拿三张图做 sanity check一张训练集里的、一张训练集外的、一张纯噪声。三张图的输出如果符合预期才继续往下调参。希望帮到你。本文还有配套的精品资源点击获取
