kasim90 commited on
Commit
50be0cf
Β·
verified Β·
1 Parent(s): fd473a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -28,18 +28,22 @@ tokenized_datasets = dataset.map(tokenize_function, batched=True)
28
 
29
  # === 4️⃣ EĞİTΔ°M AYARLARI ===
30
  # === 4️⃣ EĞİTΔ°M AYARLARI ===
 
 
 
31
  training_args = TrainingArguments(
32
  output_dir="./mistral_lora",
33
  per_device_train_batch_size=1,
34
  gradient_accumulation_steps=16,
35
  learning_rate=5e-4,
36
  num_train_epochs=1,
 
37
  save_steps=500,
38
  save_total_limit=2,
39
  logging_dir="./logs",
40
  logging_steps=10,
41
  optim="adamw_torch",
42
- no_cuda=True, # πŸ”₯ ÇÖZÜM: Ana sΓΌreΓ§te GPU'yu başlatmayΔ± engelle
43
  )
44
 
45
 
 
28
 
29
  # === 4️⃣ EĞİTΔ°M AYARLARI ===
30
  # === 4️⃣ EĞİTΔ°M AYARLARI ===
31
+ num_train_steps = 10000 // 1 # Γ–rnek başına 1 batch, toplam 10.000 veri, her batch bΓΌyΓΌklüğü 1
32
+ max_steps = num_train_steps * 1 # 1 epoch için adım sayısı (10000 veri seti için)
33
+
34
  training_args = TrainingArguments(
35
  output_dir="./mistral_lora",
36
  per_device_train_batch_size=1,
37
  gradient_accumulation_steps=16,
38
  learning_rate=5e-4,
39
  num_train_epochs=1,
40
+ max_steps=max_steps, # max_steps parametresini ekliyoruz
41
  save_steps=500,
42
  save_total_limit=2,
43
  logging_dir="./logs",
44
  logging_steps=10,
45
  optim="adamw_torch",
46
+ no_cuda=True,
47
  )
48
 
49