search

Fantasy-Studio / Paint-by-Example

Paint by Example: Exemplar-based Image Editing with Diffusion Models
https://arxiv.org/abs/2211.13227
Other
1.08k stars 96 forks source link

Minimum GPU VRAM for training #29

Open ares123-CN opened 1 year ago

ares123-CN commented 1 year ago

Amazing job, I wonder the minimum GPU VRAM for training, image size is 512 and batch size is one or two

zhangpzh commented 10 months ago

24G I guess, since I run a RTX TITAN 24G with bsz=1, it is full. OOM occurs after setting bsz>=2. By the way, I notice in the paper the authors use V100 (32G as is known to all) for training, but the config states that the bsz is 4. It puzzels me.

PuKuang commented 6 months ago

24G I guess, since I run a RTX TITAN 24G with bsz=1, it is full. OOM occurs after setting bsz>=2. By the way, I notice in the paper the authors use V100 (32G as is known to all) for training, but the config states that the bsz is 4. It puzzels me.

你好，请问你用24G 的GPU训练的时候有调整其他模型参数吗？我训练的时候32G都不够用，batch size已经是1了