溫馨提示×

怎么查看tensorflow的參數(shù)量

小億
127
2024-03-13 20:48:23
欄目: 深度學習

要查看TensorFlow模型的參數(shù)量，可以使用以下代碼片段：

import tensorflow as tf

model = tf.keras.applications.ResNet50() # 以ResNet50為例
model.summary()

運行這段代碼后，會輸出模型的結(jié)構(gòu)信息，包括每一層的名稱、輸出形狀和參數(shù)量等信息。可以根據(jù)參數(shù)量來大致估計整個模型的參數(shù)量。

0 踩