SyferText icon indicating copy to clipboard operation
SyferText copied to clipboard

BERT MLM Support

Open ArjunParthasarathy opened this issue 4 years ago • 1 comments

Description

@AlanAboudib this adds support for a BERT encoder and iterator, specifically for the Masked LM use case.

Affected Dependencies

Now requires HuggingFace Transformers library to be installed.

How has this been tested?

I used a structure very similar to the BPTT Example Notebook to verify that my encoder and iterator work in training a BERT model. I was able to get good test and validation scores for my trained model on the Wikitext-2 dataset.

ArjunParthasarathy avatar Mar 28 '21 07:03 ArjunParthasarathy

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB