I built a POS tagger and it’s having trouble learning (which is anticipated given the small dataset and large number of tags). To help, I’d like to remove the padding dimension from the output layer, so it stops predicting padding as one of the tags. Right now, pad_idx=0 and I’m wondering if there’s an easy (and reusable) way to always make the pad_idx the last element in the vocabulary so that I can just make output_dim=len(TRG.vocab)-1? Or is there a more elegant approach to this?