[fix] Tidy up todo and trainer comments

2026-09-09 11:15:08 +08:00 · 2023-01-01 02:22:57 +00:00
parent 0119ee666b
commit e27a3eb3c7
4 changed files with 22 additions and 6 deletions
@@ -92,7 +92,7 @@ class RankTrainer(Trainer):

 if __name__ == "__main__":
    training_conf = argument_parsing(parser)
-    
+
    model_name = training_conf['model_name']
    model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=1, problem_type='regression')
    if 'freeze_layer' in training_conf: