（6-3-03）CLIP模型训练与微调（3）训练模型+模型微调+调试运行

最新推荐文章于 2025-05-07 10:39:57 发布

码农三叔

最新推荐文章于 2025-05-07 10:39:57 发布

阅读量664

点赞数 8

分类专栏：多模态大模型从入门到实战(数据集、训练、RAG、多模态) 文章标签：人工智能大模型微调训练模型多模态 CLIP

本文链接：https://blog.csdn.net/asd343442/article/details/142484505

版权

多模态大模型从入门到实战(数据集、训练、RAG、多模态) 专栏收录该内容

318 篇文章 ¥49.90 ¥99.00

订阅专栏

超级会员免费看

6.3.4 训练模型

文件train.py是训练 CLIP 模型的主程序，首先根据命令行参数指定的模型名称加载相应的配置文件，然后创建一个 CLIPWrapper 模型实例，并根据命令行参数初始化数据模块。接着，使用 PyTorch Lightning 的 Trainer 对象进行训练。

import yaml
from argparse import ArgumentParser
from pytorch_lightning import Trainer
from data.text_image_dm import TextImageDataModule
from models import CLIPWrapper


def main(hparams):
    config_dir = 'models/configs/ViT.yaml' if 'ViT' in hparams.model_name else 'models/configs/RN.yaml'
    with open(config_dir) as f

了解本专栏