summaryrefslogtreecommitdiff
path: root/configs/review_user.yaml
diff options
context:
space:
mode:
Diffstat (limited to 'configs/review_user.yaml')
-rw-r--r--configs/review_user.yaml31
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