Still trying to make the trainer agnostic to tagger