Model Compression Using Optimal Transport