Thanks for sharing your awesome project. When making inference only for demo purposes, do we need L_kg or only the existing L_ce? 