自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(7)
  • 收藏
  • 关注

原创 linux大模型文件下载问题

用wget或者curl的方式下载huggingface指定项目下的指定内容

2024-03-31 18:23:27 122 1

原创 YOLOv9 安全帽检测(二)

YOLOv9安全帽检测推理

2024-03-08 01:05:43 331 1

原创 YOLOv9 安全帽检测(一)

本文记录了使用YOLOv9在安全帽检测上的训练,使用ubuntu22.04,RTX4090

2024-03-07 11:18:06 1459 1

原创 多机多卡训练SD系列模型(四)

Accelerate是为PyTorch用户设计的库,旨在简化分布式训练和混合精度训练过程。它提供了一种轻松加速和扩展PyTorch训练脚本的方式,无需编写繁琐的样板代码。Accelerate的API相对简单,仅包含一个Accelerator对象类,使用户能够更轻松地利用多GPU、TPU等计算资源,同时保持对训练循环的完全控制。

2023-10-10 18:37:15 2274 2

原创 多机多卡训练SD系列模型(三)

要写了kohya-trainer的训练脚本的使用,中间具体参数需要慢慢摸索,并且需要根据不同机器来运行不同的加速配置。

2023-10-09 21:59:17 710 1

原创 多机多卡训练SD系列模型(二)

尝试使用Dreambooth的方式,所以迁到了kohya-trainer这个项目这次训练主要是因为环境中本身就已经用了lora-script的项目,很多环境是通用的,就直接从kohya_ss中拉下来的文件直接跑了,上面还是有很多比如打标,剪裁,命名的步骤,这些在后面的项目中其实都可以省略了。后面会研究用dreambooth来训练lora,以免污染到底模。

2023-09-27 18:23:39 580 1

原创 多机多卡训练SD系列模型(一)

stable diffusion作为开源的生成图的模型已经有了很多训练完的模型,训练一个大模型还是比较麻烦的,先进行lora训练,用到了这个项目还有stable diffusion的WebUI也有很多别的lora项目对kohya的脚本进行包装这是kohya的仓库这篇作为开头,只写了从收集数据、处理数据到训练lora,测试模型的整个流程是可以跑通的,后续文章中用的项目从lora-script转到了kohya-trainer,下一篇将分析kohya-trainer的参数和dreambooth方法。

2023-09-26 20:36:48 3163

空空如也

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除