Google Colabで学習が全然進まない

はじめに

深層学習を学ぶ上でとても便利なサービスであるGoogle Colab。今回は筆者が実際に体験したGoogle Colabでの罠を解説します。

起きたこと

Google Colab上で重めのデータセットを利用して学習をしようとしている時でした。 実行時間としては1エポックあたり10分、20エポック学習予定のタスクです。 エラーも何もない実行上問題のないコードを書いたはずなのにそこで起きたのは 30分経っても1エポックも学習が進まないというとんでもない学習の遅さでした。

原因

結論から言うとデータセットの格納場所が悪く、アクセスに時間がかかったためです。 筆者が格納していたパスは/content/drive以下でした。 google driveに格納しているデータセットにアクセスして学習をすると大きいデータセットだとアクセスに膨大な時間がかかってしまいます。

解決策

google driveの中ではなく/contentの中にデータセットのディレクトリなどを格納しましょう。 これで予定通りの実行時間で学習をすることができました。

おわりに

今回はGoogle Colab上であまりにも学習が進まない時のミスについて解説しました。これで1人でもこの罠から抜け出せることを祈ります。