search

BellyBeauty / MDSAM

17 stars 2 forks source link

您好，各个模型所需的最低显存为多少 #3

Open zhoujh2020 opened 1 month ago

BellyBeauty commented 1 month ago

在512x512分辨率下，batch_size设置为16 需要显存35G 在384x384分辨率下，batch_size设置为32 需要显存38G 如果你需要把batch_size设置为1的话 512x512需要3.4G, 384x384需要 2.5G