ソースを参照

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

zhouyang.xie 3 ヶ月 前
コミット
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):
         """