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

Налаштуйте 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 повертає accuracy та 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.____)
Редагувати та запускати код