how to fine tune the bleurt model for a classification task

google-research / bleurt

BLEURT is a metric for Natural Language Generation based on transfer learning.

https://arxiv.org/abs/2004.04696

Apache License 2.0

697 stars 85 forks source link

Closed lytum closed 3 years ago

lytum commented 3 years ago

may I load the BLEURT model and add extra FC layer for a classification task?

tsellam commented 3 years ago

Hello, You could fork the code and edit model.py. Thanks!