1

我有一个扩展 Pytorch Lightning 的 LightningModule 类的类,并且有一个字段 bert_model。我正在尝试执行以下操作

self.bert_model = BertForTokenClassification.from_pretrained('bert-base-uncased')

但是,在执行此行之后,字段 bert_model 为 None。如果我这样做:

bert_model = BertForTokenClassification.from_pretrained('bert-base-uncased')
self.bert_model = bert_model

变量 bert_model 已正确初始化,而类字段 bert_model 仍为 None。

关于可能是什么问题的任何建议?

4

0 回答 0