Q&A
動画をCNNにかけてフレームごとの特徴を抽出して、そのあとLSTMを適用するモデルをtensorflowで作ったのですが、バッチサイズ分の動画を入力したらGPUのメモリが足りなくなってしまいます。
このような場合にどう対処すればバッチ学習ができるのでしょうか？
動画をバッチサイズ分だけ1つずつ入力して損失を計算するときだけtf.reducemeanを使えば良いのでしょうか？
回答1件
0
ベストアンサー
下記のように必要な分だけメモリを確保する設定をしても状況は変わりませんか？
Python: Keras/TensorFlow で GPU のメモリを必要な分だけ確保する
投稿2017/12/10 11:24
総合スコア1170
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。