train: Use closed vocab