Switch the lpips computation from cpu to cuda. ~15x faster for evaluation.

[ingra14m]

lpipsPyTorch cannot move to cuda for computation.

The new code needs additional package lpips

pip install lpips

[Gaozhongpai]

lpipsPyTorch cannot move to cuda for computation.

The new code needs additional package lpips

pip install lpips

lpipsPyTorch actually is on cuda. The problem is repeatedly loading the model.

[ingra14m]

Thanks for pointing that out.