torchbenchmark/models/BERT_pytorch/test.py (4 lines of code) (raw):

import unittest from bert_pytorch import BERT class BERTVocabTestCase(unittest.TestCase): pass