kyzhouhzau / NLPGNN

1. Use BERT, ALBERT and GPT2 as tensorflow2.0's layer. 2. Implement GCN, GAN, GIN and GraphSAGE based on message passing.
MIT License
331 stars 65 forks source link

can not import bert model #7

Open JiahuiSophieHU opened 3 years ago

JiahuiSophieHU commented 3 years ago

Hi, I can not import call bert = bert.BERT(). How can I fix it? Thanks,

bert=bert.BERT() Traceback (most recent call last): File "", line 1, in File "/home/sophie/NLPGNN/nlpgnn/models/bert.py", line 31, in init self.maxlen = param.get("maxlen", maxlen)