2022年7月10日
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) 推荐(0) 编辑