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