[feature] added summary quality rater

2026-08-09 11:40:53 +08:00 · 2023-01-01 10:19:25 +00:00
parent 1197dccf11
commit 168e9ca6b3
8 changed files with 251 additions and 15 deletions
@@ -69,6 +69,7 @@ def argument_parsing(parser):
        'eval_steps': 500,
        'loss': 'rank',
        'max_length': 440,
+        'per_device_eval_batch_size': 5,
        'per_device_train_batch_size': 8,
        'gradient_accumulation_steps': 8,
        'gradient_checkpointing': False,