Memory issue when calculating FID

IDEA-Research / HumanSD

[ICCV 2023] The official implementation of paper "HumanSD: A Native Skeleton-Guided Diffusion Model for Human Image Generation"

Apache License 2.0

271 stars 18 forks source link

Memory issue when calculating FID #20

Closed baeckgoo closed 10 months ago

baeckgoo commented 11 months ago

Thank you for sharing your great work!

I'd like to calculate FID given source images and generated images.

In your paper, the number of validation images is 3,750 images. In my experiment, the number of images is 3000, less than your case. When I calculate FID, out of memory issue happened on V100 gpu in here

Could you please give me any comments to solve it?

juxuan27 commented 11 months ago

Hi, @baeckgoo . Thank you for your focus! I haven't encounter this issue but maybe you can try whether change hyper-parameters such as fid_model_feature will help.