Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

dreambooth显存占用 #585

Open
lcc157 opened this issue Nov 21, 2024 · 0 comments
Open

dreambooth显存占用 #585

lcc157 opened this issue Nov 21, 2024 · 0 comments

Comments

@lcc157
Copy link

lcc157 commented Nov 21, 2024

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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant