Trying to make the trainer agnostic to tagger