fix typo in reward_modeling.py (huggingface#494)

BarleyXu · Jul 4, 2023 · cbb98da · cbb98da
1 parent a86eaab
commit cbb98da
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/stack_llama/scripts/reward_modeling.py b/examples/stack_llama/scripts/reward_modeling.py
@@ -41,7 +41,7 @@ class ScriptArguments:
     per_device_eval_batch_size: Optional[int] = field(default=1)
     gradient_accumulation_steps: Optional[int] = field(default=1)
     learning_rate: Optional[float] = field(default=2e-5)
-    weight_decay: Optional[int] = field(default=0.001)
+    weight_decay: Optional[float] = field(default=0.001)
     model_name: Optional[str] = field(
         default="gpt2",
         metadata={