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