تعداد پارامترها در تنسورفلو

Question

سلام.
آیا برا محاسبه ی تعداد پارامترهای شبکه عمیق و تعداد عملیات هایی که انجام میشه (فلاپس) دستور یا روش خاصی در تنسورفلو وجود داره؟؟ مثلا در کراس تعداد پارامترهای هر لایه رو میده در تنسورفلو هم چنین آپشنی داریم؟؟

Answer 1

سلام
تعداد پارامترها رو میتونید با یه تابع ساده بدست بیارید

import tensorflow as tf
def count_params():
total_parameters = 0
for variable in tf.trainable_variables():
    shape = variable.get_shape()
    variable_parameters = 1
    for dim in shape:
        variable_parameters *= dim.value
    total_parameters += variable_parameters
print("This model has %d trainable parameters"% (total_parameters))

Comments

ممنون بابت پاسختون