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") 原文档：htt
posted @ 2022-07-10 20:57  山…隹  阅读(414)  评论(0)  编辑  收藏  举报