WB/GWB Model Fusion and LMC of NNs

This repo provides code for paper 'Wasserstein Barycenter-based Model Fusion and Linear Mode Connectivity of Neural Networks'.