tfra.dynamic_embedding.get_model_mode View source on GitHub get model mode. tfra.dynamic_embedding.get_model_mode() Returns: A string: train or 'inference'