はじめに huggingfaceのTrainerクラスはhuggingfaceで提供されるモデルの事前学習のときに使うものだと思ってて、下流タスクを学習させるとき(Fine Tuning)は普通に学習のコードを実装してたんですが、下流タスクを学習させるときもTrainerクラスは使えて、めちゃくちゃ便利でした。 ただTrainerクラスのinitやTrainingArgumentsの引数はたくさんあるしよくわからん、という人のために、TrainerクラスのFine Tuning時の使い方を実装を通してまとめてみようと思います。 今回は自然言語処理のタスクとしてlivedoorニュースコーパスのタイトル文のカテゴリー分類問題をFine Tuningの例題として扱おうと思いますが、ViTのFine Tuningとかでも同様かと思います。 基本的にはhuggingfaceのTrainerクラスの
![huggingfaceのTrainerクラスを使えばFineTuningの学習コードがスッキリ書けてめちゃくちゃ便利です - Qiita](https://cdn-ak-scissors.b.st-hatena.com/image/square/8f735c3416a3bdda87a81e3c030da0775590a800/height=288;version=1;width=512/https%3A%2F%2Fqiita-user-contents.imgix.net%2Fhttps%253A%252F%252Fcdn.qiita.com%252Fassets%252Fpublic%252Farticle-ogp-background-412672c5f0600ab9a64263b751f1bc81.png%3Fixlib%3Drb-4.0.0%26w%3D1200%26mark64%3DaHR0cHM6Ly9xaWl0YS11c2VyLWNvbnRlbnRzLmltZ2l4Lm5ldC9-dGV4dD9peGxpYj1yYi00LjAuMCZ3PTk3MiZoPTM3OCZ0eHQ9aHVnZ2luZ2ZhY2UlRTMlODElQUVUcmFpbmVyJUUzJTgyJUFGJUUzJTgzJUE5JUUzJTgyJUI5JUUzJTgyJTkyJUU0JUJEJUJGJUUzJTgxJTg4JUUzJTgxJUIwRmluZVR1bmluZyVFMyU4MSVBRSVFNSVBRCVBNiVFNyVCRiU5MiVFMyU4MiVCMyVFMyU4MyVCQyVFMyU4MyU4OSVFMyU4MSU4QyVFMyU4MiVCOSVFMyU4MyU4MyVFMyU4MiVBRCVFMyU4MyVBQSVFNiU5QiVCOCVFMyU4MSU5MSVFMyU4MSVBNiVFMyU4MiU4MSVFMyU4MSVBMSVFMyU4MiU4MyVFMyU4MSU4RiVFMyU4MSVBMSVFMyU4MiU4MyVFNCVCRSVCRiVFNSU4OCVBOSVFMyU4MSVBNyVFMyU4MSU5OSZ0eHQtY29sb3I9JTIzMjEyMTIxJnR4dC1mb250PUhpcmFnaW5vJTIwU2FucyUyMFc2JnR4dC1zaXplPTU2JnR4dC1hbGlnbj1sZWZ0JTJDdG9wJnM9YmQ0ZjZiOTg1NWEwYTk2YmRmYzMyZTg1MjU1NTdjNTE%26mark-x%3D142%26mark-y%3D57%26blend64%3DaHR0cHM6Ly9xaWl0YS11c2VyLWNvbnRlbnRzLmltZ2l4Lm5ldC9-dGV4dD9peGxpYj1yYi00LjAuMCZoPTc2Jnc9NzcwJnR4dD0lNDBtX19rJnR4dC1jb2xvcj0lMjMyMTIxMjEmdHh0LWZvbnQ9SGlyYWdpbm8lMjBTYW5zJTIwVzYmdHh0LXNpemU9MzYmdHh0LWFsaWduPWxlZnQlMkN0b3Amcz1jMDQ1ZWQ4MTg0NDg4ZDExNjBjZmFlYTE4YjlkNDM0YQ%26blend-x%3D142%26blend-y%3D486%26blend-mode%3Dnormal%26s%3D0066dfe1707f37e6b3c1c297a3d9de84)