search

GaParmar / clean-fid

PyTorch - FID calculation with proper image resizing and quantization steps [CVPR 2022]
https://www.cs.cmu.edu/~clean-fid/
MIT License
971 stars 74 forks source link

Compute FID from generator doesn't fully respect num_gen #32

Closed justinpinkney closed 2 years ago

justinpinkney commented 2 years ago

Looks like the actual number of generations is rounded up to the nearest multiple of batch size:

https://github.com/GaParmar/clean-fid/blob/fca67180659eae81d2ea207b11e12acd84735171/cleanfid/fid.py#L199

GaParmar commented 2 years ago

Thank you for pointing this out! It should be correct in the recent version.

Cheers, Gaurav