Google Colaboratory上でBERTを使い、日本語文章の二値分類を行う - Qiita

この記事では、Google ColaboratoryのGPUを使用してBERTを動かし、日本語文章の二値分類を行います。 前回の記事の続きです。 詳細な説明は下記の記事で行なっていますので、初めての方に関しては、そちらを参考に進めてください。 Google Colaboratory(GPU)上でBERTのチュートリアルを行う 学習済みモデルは、GoogleDriveに保存し、GoogleColaboratoryにGoogleDriveをマウントして使用します。 ＊注意 GoogleDriveに学習済みモデルとデータセットを保存するため、1.6GB以上の空きが必要です。 一度だけ試す場合は、GoogleDriveとGoogleColaboratory、を連携せず、GoogleColaboratoryのGPUインスタンス上にモデルやデータセットを保存する方が良いかもしれません。 ＊宣伝 新しく