ПочатиПочніть безкоштовно

Ініціалізація винагороди

Ви на фінальному етапі розгортання генеративної моделі, створеної для персоналізованих рекомендацій інтернет-книгарні. Щоб узгодити цю модель з людськими вподобаннями, вам потрібно натренувати модель винагороди на зібраних даних про переваги. Перший крок — ініціалізувати модель і параметри конфігурації.

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=____)
Редагувати та запускати код