|
@@ -29,7 +29,7 @@ class ModelTrainer:
|
|
|
self.gpu_memory_utilization = config.gpu_memory_utilization
|
|
|
# 初始化 BERT 模型和分词器
|
|
|
self.tokenizer = LongformerTokenizer.from_pretrained(f'../models/allenai/longformer-base-4096')
|
|
|
- self.bert_model = LongformerModel.from_pretrained(f'../models/allenai/longformer-base-4096')
|
|
|
+ self.longformer_model = LongformerModel.from_pretrained(f'../models/allenai/longformer-base-4096')
|
|
|
|
|
|
def load_model(self):
|
|
|
"""
|