テンソルフローモデルのメモリ要件を見積もる

Question

テンソルフローモデルのメモリ要件をどのように見積もることができますか？以下は、いくらか正確な表現を与えるべきか？

size = 0 
for variable in tf.all_variables(): 
    size += int(np.prod(variable.get_shape())) 

print(size) 

sizeは変数の数である必要があります。 size * dtypeは、メモリ要件の見積もりになりますか？

Answer 1

はありません、また、あなたの他のテンソル（例えばtf.placeholderとtf.constant）を考慮する必要があり、私はそのバックプロパゲーション」はdoesnのように、値の束が往路時にキャッシュされていると信じて、あなたはまた、グラデーションのための部屋を持っている必要があります遅すぎる。