cpt core 4
Browse files
scripts/cpt_core_model_4.py
CHANGED
@@ -105,8 +105,8 @@ trainer = UnslothTrainer(
|
|
105 |
|
106 |
# learning_rate=5e-5,
|
107 |
# embedding_learning_rate=5e-6,
|
108 |
-
learning_rate = 5e-5
|
109 |
-
embedding_learning_rate = 5e-5 /
|
110 |
|
111 |
fp16=not is_bfloat16_supported(),
|
112 |
bf16=is_bfloat16_supported(),
|
|
|
105 |
|
106 |
# learning_rate=5e-5,
|
107 |
# embedding_learning_rate=5e-6,
|
108 |
+
learning_rate = 5e-5,
|
109 |
+
embedding_learning_rate = 5e-5 / 10.0,
|
110 |
|
111 |
fp16=not is_bfloat16_supported(),
|
112 |
bf16=is_bfloat16_supported(),
|