2
TensorFlowでは、ネットワーク内の学習パラメータの量を調べるために何かできることはありますか?CNN学習可能パラメータの数 - Python/TensorFlow
TensorFlowでは、ネットワーク内の学習パラメータの量を調べるために何かできることはありますか?CNN学習可能パラメータの数 - Python/TensorFlow
機能なし私は承知していますが、あなたはまだ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)
それを手に入れたのforループを使用して自分自身をカウントすることができます!ありがとうございました!そして、そのコードを置くのに最適な場所は何でしょうか? TensorFlowセッションを初期化したら? – QuestionsStackOverflow
セッションinitの前に置くことができます。グラフ変数のみを読み込み、tfセッションは必要ありません。 – Pop
ありがとうございました! – QuestionsStackOverflow