diff options
Diffstat (limited to 'configs/review_user.yaml')
| -rw-r--r-- | configs/review_user.yaml | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/configs/review_user.yaml b/configs/review_user.yaml new file mode 100644 index 0000000..d2423ee --- /dev/null +++ b/configs/review_user.yaml @@ -0,0 +1,31 @@ +task: review +setting: user +dataset_config: product_review_user +model_name: Qwen/Qwen2.5-1.5B-Instruct +device: cuda:1 + +# CVH hyperparams +d: 64 +alpha: 0.1 +beta: 0.05 +lam: 0.0001 +lr: 0.05 +adapt_steps: 30 +max_grad_norm: 5.0 +basis_seed: 42 + +# Support set +K: 4 + +# Generation +max_new_tokens: 512 +temperature: 0.0 +top_p: 0.9 + +# Evaluation +num_eval: -1 # -1 means all +batch_size: 1 + +# Blended generation +blend_gamma: 0.5 +min_new_tokens: 128 |
