mtasic85 commited on
Commit
aa7c767
·
1 Parent(s): c152e40

cpt core 4

Browse files
Files changed (1) hide show
  1. scripts/cpt_core_model_4.py +2 -2
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 * 2,
109
- embedding_learning_rate = 5e-5 / 2,
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(),