mtasic85 commited on
Commit
404ab86
·
1 Parent(s): 6304c7f

cpt core 4

Browse files
Files changed (1) hide show
  1. scripts/cpt_core_model_4.py +1 -1
scripts/cpt_core_model_4.py CHANGED
@@ -97,7 +97,7 @@ trainer = UnslothTrainer(
97
  # gradient_accumulation_steps=64,
98
  # per_device_train_batch_size=16,
99
  # gradient_accumulation_steps=16,
100
- per_device_train_batch_size=1,
101
  # gradient_accumulation_steps=8,
102
 
103
  warmup_ratio=0,
 
97
  # gradient_accumulation_steps=64,
98
  # per_device_train_batch_size=16,
99
  # gradient_accumulation_steps=16,
100
+ per_device_train_batch_size=2,
101
  # gradient_accumulation_steps=8,
102
 
103
  warmup_ratio=0,