docs, logic: обновление документации и автодовосстановления обучения модели, актуализация index.md

simple_llm/transformer/callback/lrs_scheduler_callback.py

@@ -14,8 +14,22 @@ class LRSchedulerCallback(Callback):
     def __init__(self, lr, decay=0.95):
         self.base_lr = lr
         self.decay = decay
+        self.last_epoch = -1  # Добавляем отслеживание эпохи
+
+    def get_state(self):
+        return {
+            'base_lr': self.base_lr,
+            'decay': self.decay,
+            'last_epoch': self.last_epoch
+        }
+        
+    def set_state(self, state):
+        self.base_lr = state['base_lr']
+        self.decay = state['decay']
+        self.last_epoch = state['last_epoch']
     
     def on_epoch_begin(self, epoch, model):
+        self.last_epoch = epoch  # Сохраняем текущую эпоху
         new_lr = self.base_lr * (self.decay ** epoch)
         for param_group in model.optimizer.param_groups:
             param_group['lr'] = new_lr