Model Hyper Parameters to change after pretraining on the custom dataset

[aswin-giridhar]

I had run the pretraining code of bert on a custom dataset and now i want to know which arguments i should change based on the pretrained model. The only arguments which I have changed among the three arguments(vocab_file,config_file,init_checkpoint) is the init_checkpoint which I have given the latest checkpoint created by the pretraining code. But when I tried to run it I was getting the following error.
.

So i tried changing the vocab_size in bert_config.json and tried to run it. This is the error which I am getting now.

Could you tell me the reason why am getting this issue?

[aswin-giridhar]

Working after i reduce batch_size. for further info refer issue no #92 and #82 for further info