tensorflow2.0--计算模型的时间复杂度（计算量） 

from keras_flops import get_flops

model = ...

# Calculae FLOPS
flops = get_flops(model, batch_size=1)


print(f"FLOPS: {flops / 10 ** 9:.03} G")

 

原文档：https://pypi.org/project/keras-flops/
 
posted @ 2022-07-10 20:57  山…隹  阅读(512)  评论(0)    收藏  举报