lpips-j – Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.
Project description
LPIPS-J
This is a minimal JAX/Flax port of lpips
, as implemented in:
Only the essential features have been implemented. Our motivation is to support VQGAN training for DALL•E Mini.
It currently supports the vgg16
backend, leveraging the implementation in flaxmodels
.
Pre-trained weights for the network and the linear layers are downloaded from the 🤗 Hugging Face hub.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
lpips-j-0.0.1.tar.gz
(7.1 kB
view hashes)