tfra.dynamic_embedding.enable_train_mode View source on GitHub enable train mode. tfra.dynamic_embedding.enable_train_mode()