/
githubmirror
/
agents
Обзор
Документация
Войти
/
githubmirror
/
agents
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
examples/math_dataset/run_train.py
21 строка
609 B
Jialong Wu
refactor: update relative file paths
25 сен 2024, 18:22
25 сен 2024, 18:22
f6fd0b6
Код
Авторство
О чём код?
import litellm import os from agents import HotpotQADataset, LLMConfig, OpenAILLM, Solution, SolutionConfig, MATHDataset from agents.optimization.trainer import Trainer, TrainerConfig from agents.optimization.utils import OptimUtils os.environ["OPENAI_API_KEY"] = "" os.environ["OPENAI_BASE_URL"] = "" litellm.set_verbose = False if __name__ == "__main__": # 准备数据 dataset = MATHDataset(split="train") # Trainer训练 trainer_config_path = "configs/trainer_config.json" trainer = Trainer(config=TrainerConfig( trainer_config_path), dataset=dataset) trainer.train()