mirror of
https://github.com/pese-git/simple-llm.git
synced 2026-01-24 05:26:02 +00:00
20 lines
634 B
Python
20 lines
634 B
Python
|
"""
|
||
|
|
Callback-система для управления обучением GPT.
|
||
|
|
|
||
|
|
Доступные callback-и:
|
||
|
|
- EarlyStoppingCallback - ранняя остановка
|
||
|
|
- ModelCheckpointCallback - сохранение чекпоинтов
|
||
|
|
- LRSchedulerCallback - регулировка learning rate
|
||
|
|
"""
|
||
|
|
|
||
|
|
from .callback import Callback
|
||
|
|
from .early_stopping_callback import EarlyStoppingCallback
|
||
|
|
from .lrs_scheduler_callback import LRSchedulerCallback
|
||
|
|
from .model_checkpoint_callback import ModelCheckpointCallback
|
||
|
|
|
||
|
|
__all__ = [
|
||
|
|
'Callback',
|
||
|
|
'EarlyStoppingCallback',
|
||
|
|
'LRSchedulerCallback',
|
||
|
|
'ModelCheckpointCallback'
|
||
|
|
]
|