Artificial Intelligence, Machine Learning, Bert, Healthcare, AI.
label2id, id2label = getConfig(label_list) model.config.label2id = label2id model.config.id2label = id2label In our case, we primarily want a fast training to stay under our 5 minutes mark (tic tac tic tac ⌛). So, we have chosen a learning rate of 1e-4, a number of epochs of 10 and a batch size of