Просмотр исходного кода

遵循面向对象思想重构train_model_grpo.py

zhouyang.xie 3 месяцев назад
Родитель
Сommit
c7b5cc14c5
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      src/train_model_grpo_v1.1.py

+ 1 - 1
src/train_model_grpo_v1.1.py

@@ -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):
         """