`dataset_bert_pretrained()` has some concepts that should probably be implemented (or at least matched) in `dataset_bert()`.
dataset_bert_pretrained()has some concepts that should probably be implemented (or at least matched) indataset_bert().