最近AI作畫很火，也有各種模型不斷推出，這些模型基本上都是網(wǎng)友訓(xùn)練的，而背后的技術(shù)其實(shí)也是公開的，如下各種風(fēng)格的模型層出不窮：

而本著讓這些看著高大上的技術(shù)成為傻瓜式操作的目標(biāo)，我就做了這款訓(xùn)練工具，目標(biāo)是集成各種與AI訓(xùn)練相關(guān)的功能，解壓即用，無(wú)需各種環(huán)境配置。

V3.0版本：2023-5-2

1、DB、LORA訓(xùn)練參數(shù)設(shè)置分類設(shè)置，可設(shè)置參數(shù)更加詳細(xì)；

2、支持摳圖、BLIP打圖片描述；

3、支持lora合并和提取；

4、基于torch2.0，較舊版提速1倍以上；

5、英偉達(dá)6G顯卡即可訓(xùn)練lora模型；

6、支持加載上次的參數(shù)設(shè)置、記住上次窗口大小等功能。

01?軟件概覽

軟件名字叫作：AI訓(xùn)練工具離線版V3.0，基于GitHub上的開源項(xiàng)目sd scripts。??如下，軟件主界面，還是以操作簡(jiǎn)單為主：

02?軟件安裝

安裝很簡(jiǎn)單，下載解壓即可。

第一步：下載壓縮包解壓。??

第二步：解壓的路徑中找到AI訓(xùn)練工具離線版V3.0.exe，雙擊打開，注意第一次加載比較慢，幾秒到幾十秒后會(huì)彈出如下主界面，注意背后的黑窗不能關(guān)，黑窗顯示一些重要信息。

03?軟件功能

功能一：LORA訓(xùn)練

1、SD模型選擇，lora訓(xùn)練其實(shí)是特殊的模型微調(diào)策略，所以要選擇SD模型，SD模型通用的一般是SD 1.5，二次元一般是novelAI、revAnimated_v11這種大模型，我演示的SD 1.5。

2、訓(xùn)練圖片參數(shù)選擇

（1）準(zhǔn)備訓(xùn)練圖及目錄：這是重點(diǎn)，主要是訓(xùn)練圖片目錄選擇，以及文件夾建立方式，如下，我準(zhǔn)備的訓(xùn)練圖片：

可以看到，圖片放到10_mingren文件夾里，10代表讓AI重復(fù)訓(xùn)練次數(shù)，mingren代表模型標(biāo)識(shí)詞，后續(xù)訓(xùn)練出的模型，描述里帶著mingren就會(huì)畫和訓(xùn)練圖風(fēng)格類似的圖片。切記嚴(yán)格這樣建立目錄，格式就是：重復(fù)次數(shù)_標(biāo)識(shí)詞，下劃線隔開。

（2）選擇訓(xùn)練圖片目錄，如下，訓(xùn)練圖片目錄只選擇到10_mingren的上層目錄訓(xùn)練圖文件夾，這一步也切記嚴(yán)格執(zhí)行。

（3）正則圖片目錄：這個(gè)主要防止過(guò)擬合，可以不選擇，正則圖片目錄和訓(xùn)練圖片目錄建立策略一致，不過(guò)重復(fù)次數(shù)1就行，目錄名字格式重復(fù)次數(shù)_標(biāo)類別詞，比如1_man。正則圖片目錄這個(gè)基本不用選，空著就行。

（4）圖片寬高：這個(gè)和訓(xùn)練圖片尺寸有關(guān)，比如訓(xùn)練圖片尺寸都是512×704，自然就是設(shè)置成寬512，高704。我的訓(xùn)練圖都是512×512的，所以我設(shè)置都是512。

（5）最小最大分辨率：默認(rèn)最小256，最大1024，代表你訓(xùn)練圖的尺寸最小邊不能小于256，最大邊不能超過(guò)1024，比如你訓(xùn)練圖某一張分辨率為512×1280，最大邊1280是大于1024的，這種會(huì)報(bào)錯(cuò)。

3、LORA模型參數(shù)

（1）模型框架：這個(gè)默認(rèn)就行，其他框架還在豐富擴(kuò)充中，后續(xù)會(huì)加入更多，目前測(cè)試默認(rèn)的效果就不錯(cuò)。

（2）LORA斷點(diǎn)模型選擇：這個(gè)就是如果發(fā)現(xiàn)訓(xùn)練出的模型沒(méi)訓(xùn)練夠，可選擇上次的lora模型繼續(xù)訓(xùn)練，如果沒(méi)有這個(gè)需求，這個(gè)自然就不用選擇了。

（3）網(wǎng)絡(luò)維度參數(shù)：network_dim參數(shù)越大，擬合效果越好，這個(gè)自己測(cè)試，可以鼠標(biāo)懸停到旁邊的問(wèn)號(hào)有解釋。

4、訓(xùn)練基本參數(shù)

（1）batch size：默認(rèn)為1，顯存大的可以設(shè)置大一些，訓(xùn)練速度會(huì)快很多。

（2）clip skip：默認(rèn)為2，這個(gè)默認(rèn)就行，基本都用2。

（3）epoch和保存頻率：默認(rèn)10個(gè)epoch，每?jī)蓚€(gè)epoch保存一次模型（保存中途模型的好處是有時(shí)候最終模型可能過(guò)擬合，可能中途的模型效果要好一些），epoch代表模型訓(xùn)練輪數(shù)，一般設(shè)置大一些擬合效果好，但太大也沒(méi)啥必要，還有個(gè)參數(shù)叫步數(shù)step，比如10張訓(xùn)練圖，重復(fù)次數(shù)10，batch size為1，訓(xùn)練10個(gè)epoch，那么step=10*10*10=1000步。

5、學(xué)習(xí)率

學(xué)習(xí)率是一個(gè)比較玄學(xué)或者說(shuō)進(jìn)階的參數(shù)，一般默認(rèn)就行了。

6、優(yōu)化器

優(yōu)化器的作用是讓AI沿著不同方向去擬合模型，有些優(yōu)化器帶著學(xué)習(xí)率自適應(yīng)調(diào)整的策略，默認(rèn)的Adam8bit就不錯(cuò)，當(dāng)然有說(shuō)Lion也不錯(cuò)的，各位自行測(cè)試。

7、其他參數(shù)

這幾個(gè)參數(shù)還是默認(rèn)就好。

8、模型保存

模型目錄名可以修改一下，這樣會(huì)保存到軟件目錄/生成的模型目錄中，保存精度默認(rèn)float32，20、30系列顯卡可以設(shè)置保存fp16，這樣保存的模型體積小一倍，效果沒(méi)區(qū)別。

設(shè)置完之后，可以點(diǎn)擊訓(xùn)練了，訓(xùn)練完之后的模型保存在軟件目錄/生成的模型目錄中。一份極簡(jiǎn)的lora設(shè)置是：默認(rèn)設(shè)置下，選擇SD模型，選擇訓(xùn)練圖片目錄，點(diǎn)擊訓(xùn)練，就三步，其他參數(shù)不用修改。

功能二：DreamBooth訓(xùn)練

DreamBooth訓(xùn)練參數(shù)包含于Lora訓(xùn)練參數(shù)設(shè)置，訓(xùn)練機(jī)理也和Lora一致，所以這里不再贅述。這里特別說(shuō)的是DreamBooth訓(xùn)練出的模型是大模型，也就是好幾個(gè)GB大小的，而Lora一般幾十MB到上百M(fèi)B大小。

功能三：圖片批量描述打標(biāo)簽

不管是Lora訓(xùn)練還是DreamBooth訓(xùn)練，對(duì)圖片打標(biāo)簽也挺重要，如下點(diǎn)擊點(diǎn)擊綠色按鈕，選擇圖片所在文件夾，即可自動(dòng)生成圖片同名的txt文件與圖片同級(jí)目錄保存，支持文件夾下的子文件夾圖片打標(biāo)簽。

功能四：批量摳圖

對(duì)于訓(xùn)練諸如人像，扣除背景也挺有必要，如下點(diǎn)擊綠色按鈕，批量選擇要扣除的圖片，即可自動(dòng)摳圖，保存到軟件目錄\images_out\Koutu_Out文件夾里，當(dāng)然畢竟AI自動(dòng)摳圖，有些圖扣的效果有一些問(wèn)題。▼

功能五：LORA融合

lora模型和SD模型一樣，都支持按比例融合，比如一個(gè)二次元和一個(gè)三次元模型一比一融合，融合出來(lái)的模型可能是2.5次元風(fēng)格的。如下界面，選擇要融合的lora模型，至少選擇兩個(gè)，設(shè)置比例，點(diǎn)擊融合即可。

功能六：LORA提取

這個(gè)功能有點(diǎn)意思，就是從網(wǎng)上的SD模型里提取出Lora模型，基底模型就是這個(gè)大模型基于哪個(gè)模型訓(xùn)練的，一般是SD 1.5，最終模型就是網(wǎng)上下的SD模型，然后選擇保存目錄和lora dim，點(diǎn)擊開始提取即可。這個(gè)很玄學(xué)，因?yàn)槟悴恢谰W(wǎng)上下載模型基底模型到底用的哪個(gè)，各位可以測(cè)測(cè)，不同的基底模型提取的Lora沒(méi)準(zhǔn)有奇效。

04?配置要求

需要英偉達(dá)顯卡，6G顯存顯卡，Windows 10、11系統(tǒng)電腦，像1060、980ti、1660、2060等6G顯卡都可以，麗臺(tái)、tesla計(jì)算卡等都支持，基本上就是英偉達(dá)6G顯存顯卡就行了。

05?小結(jié)

我軟件名字叫做AI訓(xùn)練軟件，就代表后續(xù)加入的訓(xùn)練功能不局限于作畫模型的訓(xùn)練，諸如小說(shuō)、音視頻相關(guān)的訓(xùn)練功能后續(xù)都會(huì)加入，這個(gè)項(xiàng)目確實(shí)很有意思，感興趣的可以體驗(yàn)一下3.0版本了，更詳細(xì)的教程多刷刷B站相關(guān)視頻即可。

????Windows 10、Windows11版????下載

方式一：百度網(wǎng)盤
https://pan.baidu.com/s/1nJD2wDWyAcNBkbD7ThQlMQ?pwd=vjc8
方式二：天翼網(wǎng)盤
https://cloud.189.cn/t/iEB7nqyq2a6z
（訪問(wèn)碼：ca02）