search

cnulab / RealNet

Offical implementation of "RealNet: A Feature Selection Network with Realistic Synthetic Anomaly for Anomaly Detection (CVPR 2024)"
MIT License
143 stars 5 forks source link

请问训练diffusion model 需要多大的显存 #2

Closed leefty-1024 closed 3 months ago

leefty-1024 commented 3 months ago

我对diffusion model了解不多，请问训练diffusion model 大概需要多大的显存，感谢您！

cnulab commented 3 months ago

在256*256的分辨率下，最少需要单卡24G显存，24G显存只能设置batch size=1，我推荐你使用单卡48G显存，可设置batch size=5，单卡80G显存可设置batch size=10，如果你仅想复现结果，可以使用我提供的diffusion checkpoints.