BioMegatron token classification tutorial fix to be compatible with current Megatron BERT
The token classification model uses the Megatron BERT model loaded from nemo file, which is converted from the Megatron-LM checkpoints. 

Currently it fails to reproduce the same accuracy. Need to investigate the bug. 