cpt core 4
Browse files
scripts/cpt_core_model_4.py
CHANGED
@@ -121,15 +121,15 @@ trainer = UnslothTrainer(
|
|
121 |
output_dir=output_dir,
|
122 |
report_to='wandb',
|
123 |
|
124 |
-
save_steps=1000,
|
125 |
run_name=run_name,
|
|
|
126 |
|
127 |
do_eval=True,
|
128 |
fp16_full_eval=True,
|
129 |
per_device_eval_batch_size=1,
|
130 |
eval_accumulation_steps=4,
|
131 |
eval_strategy='steps',
|
132 |
-
eval_steps=
|
133 |
),
|
134 |
)
|
135 |
|
|
|
121 |
output_dir=output_dir,
|
122 |
report_to='wandb',
|
123 |
|
|
|
124 |
run_name=run_name,
|
125 |
+
save_steps=1000,
|
126 |
|
127 |
do_eval=True,
|
128 |
fp16_full_eval=True,
|
129 |
per_device_eval_batch_size=1,
|
130 |
eval_accumulation_steps=4,
|
131 |
eval_strategy='steps',
|
132 |
+
eval_steps=1000,
|
133 |
),
|
134 |
)
|
135 |
|