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

fix: pass on device to DataParallel; line breaks. #65

Open janfb opened 4 months ago

janfb commented 4 months ago

fixes #60

Changes:

Note that I added docstrings and types for all functions in fid.py. Given that so many people are using the library I thought it would be helpful to give a bit more context for the users. Do let me know though, if you prefer not having theses changes, or having them in a separate PR.

Best, Jan