logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

SAM2 微调代码详解-----dataloader

官方提供的微调代码是在MOSE数据集上进行微调,如下所示:由于我暂未下载MOSE数据集,所以采用了与MOSE数据集结构相似的DAVIS数据集。有关Hydra的知识可参考:我们首先来看data,这涉及数据是如何导入的,如果我们要使用自己的数据集进行微调,我们需要进行哪些数据预处理。这是一个层层嵌套的过程,从而也加大了我们的阅读难度,我们从最底层的类开始逐步解析。

文章图片
#深度学习#人工智能#计算机视觉
transformers库与DDP多卡训练冲突

这时候,在每个进程里跑的都是一个模型副本,用的是单 GPU,这和 DDP 是兼容的。而 DDP 是把同一个模型的多个副本分别放在不同的 GPU 上,靠梯度同步。device_map="auto"是 HuggingFace 提供的一种。,改成手动 .to(f"cuda:{local_rank}")。,它会自动将大型模型。

#transformer#深度学习#人工智能
山东大学创新实训第七周周报-----VisualGLM-6B微调环境配置

需要更新GPU 驱动程序,以便与 PyTorch 的版本匹配,或更新PyTorch 的版本,以便与 GPU 驱动程序匹配。找到下载的14个G的那个visualglm-6b.zip,然后unzip,然后解压的文件里面有个model_config.json文件,里面。可能依赖的transformers模块版本不匹配,导致了tokenizer对象的属性不一致。在当前的 PyTorch 版本中,由于我们所

文章图片
#语言模型#人工智能#计算机视觉
到底了