├── LICENSE ├── README.md ├── assets ├── speedup.png └── vis.png ├── engine.py ├── environment_setup.sh ├── fid_stats ├── adm_in256_stats.npz └── adm_in512_stats.npz ├── main.py ├── main_cache.py ├── models ├── block.py ├── lpd.py ├── mask.py ├── sampling.py └── vqgan.py ├── orders ├── group_sizes_lookup.json ├── group_sizes_training │ ├── group_sizes_256.json │ └── group_sizes_512.json ├── lpd_order.py ├── run_lpd_order.py └── run_lpd_order.sh ├── requirements.txt ├── scripts ├── cli │ └── run.py ├── eval │ ├── lpd_l_res256_steps20.sh │ ├── lpd_l_res256_steps32.sh │ ├── lpd_l_res512_steps48.sh │ ├── lpd_xl_res256_steps20.sh │ ├── lpd_xl_res256_steps32.sh │ ├── lpd_xl_res512_steps48.sh │ └── lpd_xxl_res256_steps20.sh ├── setups │ └── train.sh └── train │ ├── lpd_l_256.sh │ ├── lpd_l_512.sh │ ├── lpd_xl_256.sh │ ├── lpd_xl_512.sh │ └── lpd_xxl_256.sh └── util ├── crop.py ├── loader.py ├── lr_sched.py └── misc.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/README.md -------------------------------------------------------------------------------- /assets/speedup.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/assets/speedup.png -------------------------------------------------------------------------------- /assets/vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/assets/vis.png -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/engine.py -------------------------------------------------------------------------------- /environment_setup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/environment_setup.sh -------------------------------------------------------------------------------- /fid_stats/adm_in256_stats.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/fid_stats/adm_in256_stats.npz -------------------------------------------------------------------------------- /fid_stats/adm_in512_stats.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/fid_stats/adm_in512_stats.npz -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/main.py -------------------------------------------------------------------------------- /main_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/main_cache.py -------------------------------------------------------------------------------- /models/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/models/block.py -------------------------------------------------------------------------------- /models/lpd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/models/lpd.py -------------------------------------------------------------------------------- /models/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/models/mask.py -------------------------------------------------------------------------------- /models/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/models/sampling.py -------------------------------------------------------------------------------- /models/vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/models/vqgan.py -------------------------------------------------------------------------------- /orders/group_sizes_lookup.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/orders/group_sizes_lookup.json -------------------------------------------------------------------------------- /orders/group_sizes_training/group_sizes_256.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/orders/group_sizes_training/group_sizes_256.json -------------------------------------------------------------------------------- /orders/group_sizes_training/group_sizes_512.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/orders/group_sizes_training/group_sizes_512.json -------------------------------------------------------------------------------- /orders/lpd_order.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/orders/lpd_order.py -------------------------------------------------------------------------------- /orders/run_lpd_order.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/orders/run_lpd_order.py -------------------------------------------------------------------------------- /orders/run_lpd_order.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/orders/run_lpd_order.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/cli/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/cli/run.py -------------------------------------------------------------------------------- /scripts/eval/lpd_l_res256_steps20.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/eval/lpd_l_res256_steps20.sh -------------------------------------------------------------------------------- /scripts/eval/lpd_l_res256_steps32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/eval/lpd_l_res256_steps32.sh -------------------------------------------------------------------------------- /scripts/eval/lpd_l_res512_steps48.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/eval/lpd_l_res512_steps48.sh -------------------------------------------------------------------------------- /scripts/eval/lpd_xl_res256_steps20.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/eval/lpd_xl_res256_steps20.sh -------------------------------------------------------------------------------- /scripts/eval/lpd_xl_res256_steps32.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/eval/lpd_xl_res256_steps32.sh -------------------------------------------------------------------------------- /scripts/eval/lpd_xl_res512_steps48.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/eval/lpd_xl_res512_steps48.sh -------------------------------------------------------------------------------- /scripts/eval/lpd_xxl_res256_steps20.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/eval/lpd_xxl_res256_steps20.sh -------------------------------------------------------------------------------- /scripts/setups/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/setups/train.sh -------------------------------------------------------------------------------- /scripts/train/lpd_l_256.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/train/lpd_l_256.sh -------------------------------------------------------------------------------- /scripts/train/lpd_l_512.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/train/lpd_l_512.sh -------------------------------------------------------------------------------- /scripts/train/lpd_xl_256.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/train/lpd_xl_256.sh -------------------------------------------------------------------------------- /scripts/train/lpd_xl_512.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/train/lpd_xl_512.sh -------------------------------------------------------------------------------- /scripts/train/lpd_xxl_256.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/scripts/train/lpd_xxl_256.sh -------------------------------------------------------------------------------- /util/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/util/crop.py -------------------------------------------------------------------------------- /util/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/util/loader.py -------------------------------------------------------------------------------- /util/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/util/lr_sched.py -------------------------------------------------------------------------------- /util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/lpd/HEAD/util/misc.py --------------------------------------------------------------------------------