Update train.py

train.py

@@ -8,7 +8,7 @@ import torch
 model_name = "google/flan-t5-large"
 run_id = "flan-t5-large-ru-autobatch"
 output_dir = f"./{run_id}"
-start_batch_size = 10   # ⚠️ Начинаем с небольшого batch, чтобы избежать OOM
+start_batch_size = 20   # ⚠️ Начинаем с небольшого batch, чтобы избежать OOM
 step_batch_size = 1
 
 # 📦 Загружаем модель и токенизатор
@@ -42,7 +42,7 @@ def tokenize_function(examples):
 tokenized_datasets = dataset.map(tokenize_function, batched=True)
 
 # 🔑 Авторизация W&B
-wandb.login(key="ВАШ_WANDB_КЛЮЧ_ЗДЕСЬ")
+wandb.login(key="5f028bc0142fb7fa45bdacdde3c00dbbaf8bf98e")
 
 # 🚀 Функция автоподбора batch size
 def try_training_with_batch_size(batch_size_start):