НачатьНачать бесплатно

Настройка Trainer

Ваша модель будет заменять сложные фразы более простыми и понятными перефразировками, улучшая читаемость переводов. Например, она может упростить фразу «The inclement weather conditions precipitated the postponement of the outdoor event» до «The bad weather caused the outdoor event to be delayed.» Создайте Trainer, чтобы подготовиться к обучению сервиса языкового перевода! Обратите внимание, что выполнение упражнения займёт некоторое время из-за вызова trainer.train().

Некоторые данные уже загружены:

  • model — модель на основе Transformer
  • dataset содержит набор данных MRPC с парафразами предложений
  • функция compute_metrics возвращает точность и F1-меру
  • training_args определён в предыдущем упражнении

Это упражнение является частью курса

Эффективное обучение моделей ИИ с PyTorch

Посмотреть курс

Инструкции к упражнению

  • Передайте model в класс Trainer().
  • Передайте аргументы обучения в класс Trainer().
  • Передайте функцию для вычисления метрик в класс Trainer().
  • Выведите на экран устройство, выбранное объектом trainer.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

trainer = Trainer(
    # Pass in the model
    model=____,
    # Input the training arguments
    args=____,
    train_dataset=dataset["train"],
    eval_dataset=dataset["validation"],
    # Pass in a function to compute metrics
    compute_metrics=____,
)

trainer.train()

# Print the device that the trainer chooses
print(trainer.args.____)
Редактировать и запускать код