OMGDGAN 模型壓縮框架
OMGD 即 Online Multi-Granularity Distillation（在線多粒度蒸餾算法）的簡稱，是一款模型壓縮框架，專門用于解決 GAN 模型體積過大，過于耗費(fèi)算力的問題。和同類模型壓縮算法相比，OMGD 不僅壓得更小，而且壓得更好。
比如在把馬變成斑馬的過程中：
（MACs 表示消耗的計算量，括號中的數(shù)字為提升倍數(shù)）
把夏天變成冬天：
實驗表明，這項技術(shù)可以把 GAN 模型的計算量最低減少到原來的 1/46、參數(shù)量最低減少到原來的 1/82。
OMGD 整體 Pipeline
性能
環(huán)境要求
- Linux
- Python 3
- CPU or NVIDIA GPU + CUDA CuDNN
評論
圖片
表情