目前還只訓練了imagenet，一百多萬張圖，1000類，類別標籤生圖，後續考慮加上文本生圖
目前共有12篇帖子。
圖像生成模型訓練實驗
 |
首先訓練了VAE作為圖像壓縮模型，然後訓練DiT作為圖像生成模型
|
|
150000步採樣（batchsize=32,按照這個計算，全數據集每張圖已經過了四遍）：
悄悄打开魔盒：單張3090訓練速度非常有限，一小時只有一萬步左右，所以到現在還看不出來什麼穩定的圖像
悄悄打开魔盒：回復 @悄悄打開魔盒：初學者建議先使用手寫數字數據集MNIST或者衣服數據集Fashion-MNIST，圖片小，訓練速度快，能很快看到效果
[查看詳情]
|
|
 悄悄打开魔盒：100萬步訓練，單張3090約耗時四天。這是前16類圖像的採樣結果。
0 n01440764 魚 , tench, Tinca tinca1 n01443537 魚, goldfish, Carassius auratus 2 n01484850 魚, great white shark, white shark, man-eater, man-eating shark, Carcharodon carcharias 3 n01491361 魚, tiger shark, Galeocerdo cuvieri 4 n01494475 魚, hammerhead, hammerhead shark 5 n01496331 魚, electric ray, crampfish, numbfish, torpedo 6 n01498041 魚, stingray 7 n01514668 雞 , cock8 n01514859 雞, hen 9 n01518878 鴕鳥 , ostrich, Struthio camelus10 n01530575 鳥, brambling, Fringilla montifringilla 11 n01531178 鳥, goldfinch, Carduelis carduelis 12 n01532829 鳥, house finch, linnet, Carpodacus mexicanus 13 n01534433 鳥, junco, snowbird 14 n01537544 鳥, indigo bunting, indigo finch, indigo bird, Passerina cyanea 15 n01558993 鳥, robin, American robin, Turdus migratorius
[查看詳情]
|
|
 悄悄打开魔盒：
16 n01560419 鳥, bulbul 17 n01580077 鳥, jay 18 n01582220 鳥, magpie 19 n01592084 鳥, chickadee 20 n01601694 鳥, water ouzel, dipper 21 n01608432 鳥, kite 22 n01614925 鷹, bald eagle, American eagle, Haliaeetus leucocephalus 23 n01616318 鷹, vulture 24 n01622779 貓頭鷹 , great grey owl, great gray owl, Strix nebulosa25 n01629819 壁虎, European fire salamander, Salamandra salamandra 26 n01630670 壁虎, common newt, Triturus vulgaris 27 n01631663 壁虎, eft 28 n01632458 壁虎, spotted salamander, Ambystoma maculatum 29 n01632777 壁虎, axolotl, mud puppy, Ambystoma mexicanum 30 n01641577 蛤蟆, bullfrog, Rana catesbeiana 31 n01644373 青蛙 , tree frog, tree-frog
[查看詳情]
|
|
 悄悄打开魔盒：
32 n01644900 青蛙, tailed frog, bell toad, ribbed toad, tailed toad, Ascaphus trui 33 n01664065 龜 , loggerhead, loggerhead turtle, Caretta caretta34 n01665541 龜, leatherback turtle, leatherback, leathery turtle, Dermochelys coriacea 35 n01667114 龜, mud turtle 36 n01667778 龜, terrapin 37 n01669191 龜, box turtle, box tortoise 38 n01675722 蜥蜴 , banded gecko39 n01677366 蜥蜴, common iguana, iguana, Iguana iguana 40 n01682714 蜥蜴, American chameleon, anole, Anolis carolinensis 41 n01685808 蜥蜴, whiptail, whiptail lizard 42 n01687978 蜥蜴, agama 43 n01688243 蜥蜴, frilled lizard, Chlamydosaurus kingi 44 n01689811 蜥蜴, alligator lizard 45 n01692333 蜥蜴, Gila monster, Heloderma suspectum 46 n01693334 蜥蜴, green lizard, Lacerta viridis 47 n01694178 蜥蜴, African chameleon, Chamaeleo chamaeleon
[查看詳情]
|
|
 悄悄打开魔盒：
200 n02097474 狗, Tibetan terrier, chrysanthemum dog 201 n02097658 狗, silky terrier, Sydney silky 202 n02098105 狗, soft-coated wheaten terrier 203 n02098286 狗, West Highland white terrier 204 n02098413 狗, Lhasa, Lhasa apso 205 n02099267 狗, flat-coated retriever 206 n02099429 狗, curly-coated retriever 207 n02099601 狗, golden retriever 208 n02099712 狗, Labrador retriever 209 n02099849 狗, Chesapeake Bay retriever 210 n02100236 狗, German short-haired pointer 211 n02100583 狗, vizsla, Hungarian pointer 212 n02100735 狗, English setter 213 n02100877 狗, Irish setter, red setter 214 n02101006 狗, Gordon setter 215 n02101388 狗, Brittany spaniel
[查看詳情]
|
