2017-11-15 23 views

答えて

3

機能なし私は承知していますが、あなたはまだtf.trainable_variables():

total_parameters = 0 
for variable in tf.trainable_variables(): 
    variable_parameters = 1 
    for dim in variable.get_shape(): 
     variable_parameters *= dim.value 
    total_parameters += variable_parameters 

print("Total number of trainable parameters: %d" % total_parameters) 
+0

それを手に入れたのforループを使用して自分自身をカウントすることができます!ありがとうございました!そして、そのコードを置くのに最適な場所は何でしょうか? TensorFlowセッションを初期化したら? – QuestionsStackOverflow

+0

セッションinitの前に置くことができます。グラフ変数のみを読み込み、tfセッションは必要ありません。 – Pop

+0

ありがとうございました! – QuestionsStackOverflow

関連する問題