Akegarasu / Akegarasu/lora-scripts

dreambooth显存占用

未關閉
#585 0 則留言 0 個 reaction 已指派 0 人 在 GitHub 檢視
主要語言
Python
星號
6.1k
分支
699
PR 合併指標
30 天內沒有已合併 PR

描述

dreambooth显存占用是不是挺大的,训练一个512大小得图,4张p40 24G显卡都不够,这个正常吗

batch_size: 1
resolution: (256, 256)
enable_bucket: True
network_multiplier: 1.0
min_bucket_reso: 256
max_bucket_reso: 512
bucket_reso_steps: 64
bucket_no_upscale: False

[Subset 0 of Dataset 0]
image_dir: "lora-scripts/imgs/7_zkz/7_zkz"
image_count: 1
num_repeats: 7
shuffle_caption: True
keep_tokens: 0
keep_tokens_separator:
secondary_separator: None
enable_wildcard: False
caption_dropout_rate: 0.0
caption_dropout_every_n_epoches: 0
caption_tag_dropout_rate: 0.0
caption_prefix: None
caption_suffix: None
color_aug: False
flip_aug: False
face_crop_aug_range: None
random_crop: False
token_warmup_min: 1,
token_warmup_step: 0,
is_reg: False
class_tokens: zkz
caption_extension: .txt

貢獻指南

這個儲存庫沒有索引到貢獻指南

評估

這個 Issue 還沒有評估資料。

把新 issue 寄到你的電子郵件信箱

精選適合新手參與的 GitHub issue 摘要。