[2020-03-17 23:07:55]: bert train_bert.py[line:86] INFO ---------- Bert Train ... ----------
Traceback (most recent call last):
File "E:/IdeaProjects/CAIL2019/CAIL2019-master/train_bert.py", line 91, in
model = BertForTripletNet.from_pretrained(BERT_MODEL_PATH)
File "C:\Python37\lib\site-packages\pytorch_transformers\modeling_utils.py", line 474, in from_pretrained
model = cls(config, *model_args, **model_kwargs)
File "E:\IdeaProjects\CAIL2019\CAIL2019-master\models\net2.py", line 10, in init
self.bert = BertModel(config)
File "C:\Python37\lib\site-packages\pytorch_transformers\modeling_bert.py", line 656, in init
self.embeddings = BertEmbeddings(config)
File "C:\Python37\lib\site-packages\pytorch_transformers\modeling_bert.py", line 253, in init
self.LayerNorm = BertLayerNorm(config.hidden_size, eps=config.layer_norm_eps)
File "C:\Python37\lib\site-packages\apex-0.1-py3.7.egg\apex\normalization\fused_layer_norm.py", line 133, in init
fused_layer_norm_cuda = importlib.import_module("fused_layer_norm_cuda")
File "C:\Python37\lib\importlib_init_.py", line 127, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "", line 1006, in _gcd_import
File "", line 983, in _find_and_load
File "", line 965, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'fused_layer_norm_cuda'