Added a dummy class for kt loss aug inference.