モデル学習時のGPUメモリ不足(RuntimeError: CUDA out of memory)を解消した
こんにちは、エンジニアのすずきです。
GWは日本橋へ行くので、ひさびさに小洞天のシュウマイと五目焼きそばを食べてこようと思っています。余裕があったら台南担々麺も頼んじゃうよ。
ところで、BERTベースモデル(TabBERT)の学習時、入力データのサイズを大きくしたらGPUメモリ不足のエラーがでてしまうようになってしまいました。
AlgorithmError: ExecuteUserScriptError: ExitCode 1 ErrorMessage "RuntimeE