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
894 stars 68 forks source link

FID clip crashes when evaluated on `cpu` device #41

Closed Sumith1896 closed 1 year ago

Sumith1896 commented 1 year ago

Passing device="cpu" and model_name="clip_vit_b_32" crashes with the following error:

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument weight in method wrapper___slow_conv2d_forward)
GaParmar commented 1 year ago

Thanks for pointing this out! It should be resolved in the latest release 0.1.32