taishan1994 / pytorch_bert_multi_classification

基于pytorch_bert的中文多标签分类
79 stars 14 forks source link

报错 #4

Open ZXB1020 opened 1 year ago

ZXB1020 commented 1 year ago

Some weights of the model checkpoint at ../model_hub/bert-base-chinese/ were not used when initializing BertModel: ['cls.predictions.transform.dense.bias', 'cls.predictions. transform.LayerNorm.weight', 'cls.predictions.transform.dense.weight', 'cls.predictions.decoder.weight', 'cls.seq_relationship.weight', 'cls.seq_relationship.bias', 'cls.pre dictions.bias', 'cls.predictions.transform.LayerNorm.bias']

taishan1994 commented 1 year ago
  • AssertionError: Torch not compiled with CUDA enabled

是否装了GPU版本的pytorch,电脑上是否有GPU可用。

ZXB1020 commented 1 year ago

明白了,谢谢!