Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|
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 |
|