Ініціалізація винагороди
Ви на фінальному етапі розгортання генеративної моделі, створеної для персоналізованих рекомендацій інтернет-книгарні. Щоб узгодити цю модель з людськими вподобаннями, вам потрібно натренувати модель винагороди на зібраних даних про переваги. Перший крок — ініціалізувати модель і параметри конфігурації.
AutoTokenizer і AutoModelForSequenceClassification уже завантажено з transformers. RewardConfig завантажено з trl.
Ця вправа є частиною курсу
Reinforcement Learning from Human Feedback (RLHF)
Інструкції до вправи
- Завантажте модель GPT-1,
"openai-gpt", для задачі класифікації послідовностей за допомогоюAutoModelForSequenceClassificationвід Hugging Face. - Ініціалізуйте конфігурацію винагороди, використовуючи
"output_dir"як вихідний каталог, і встановіть максимальну довжину токенів60.
Інтерактивна практична вправа
Спробуйте виконати цю вправу, доповнивши цей зразок коду.
# Load the pre-trained GPT-1 model for text classification
model = ____
tokenizer = AutoTokenizer.from_pretrained("openai-gpt")
# Initialize the reward configuration and set max_length
config = ____(output_dir=____, max_length=____)