tensorflow2计算flops

TensorFlow2不使用第三方库计算flops，本人python3.9，TensorFlow2.5

代码如下（示例）：

import tensorflow as tf
from tensorflow.python.profiler.model_analyzer import profile
from tensorflow.python.profiler.option_builder import ProfileOptionBuilder
print('TensorFlow:', tf.__version__)
# 这里换成自己的模型
model = tf.keras.applications.ResNet50()forward_pass = tf.function(model.call,input_signature=[tf.TensorSpec(shape=(1,) + model.input_shape[1:])])graph_info = profile(forward_pass.get_concrete_function().graph,options=ProfileOptionBuilder.float_operation())# The //2 is necessary since `profile` counts multiply and accumulate
# as two flops, here we report the total number of multiply accumulate ops
flops = graph_info.total_float_ops // 2
print('Flops: {:,}'.format(flops))

tensorflow2计算flops相关推荐

pytorch计算FLOPs
转自:pytorch计算FLOPs - 简书 1. 引言其实模型的参数量好算,但浮点运算数并不好确定,我们一般也就根据参数量直接估计计算量了.但是像卷积之类的运算,它的参数量比较小,但是运算量非常大 ...
Tensor flow计算Flops
Tensorflow计算Flops: run_meta = tf.compat.v1.RunMetadata() opts = tf.compat.v1.profiler.ProfileOptionB ...
计算FLOPs和param
概念网络中的参数量(param)对应与空间Space概念,及空间复杂度. 浮点计算量(FLOPs)对应与时间Time概念,对应于时间复杂度. 如何计算网络中的参数量(param) 网络中的参数量(p ...
tensorflow计算flops
with tf.Session() as sess: run_meta = tf.RunMetadata() opts = tf.profiler.ProfileOptionBuilder.float ...
CNN 模型的参数（parameters）数量和浮点运算数量（FLOPs）是怎么计算的
文章目录: 1 模型参数(parameters)的个数 1.1 模型参数介绍 1.2 常见网络的模型参数个数 1.2 举例如何计算神经网络模型参数 2 FLOPS:每秒浮点运算次数 2.1 `FLOP ...
pytorch 模型可视化_PyTorch Tips(FLOPs计算/参数量/计算图可视化/相关性分析)
最近在做Research Project的时候,发现有些小工具很好用,记录在此. 1. 准确的FLOPS 计算网上开源的很多计算flops的工具只支持计算PyTorch内置层的flops,不能有效计 ...
PyTorch中FLOPs计算问题
最近看了很多关于FLOPs计算的实现方法,也自己尝试了一些方法,发现最好用的还是PyTorch中的thop库(代码如下): device = torch.device("cuda" ...
深度学习中FLOPs计算
深度学习中FLOPs计算定义:FLOPs(floating point operations),即浮点数运算次数,是一个计算量,所以可以用来衡量一个算法/模型等的复杂度 FLOPs是用来计算整个网络 ...
面试宝典笔记：卷积计算过程中的FLOPs
模型的运算次数,可用 FLOPs衡量,也就是浮点运算次数(FLoating-point OPerations),表征的是模型的时间复杂度.模型空间复杂度通过Parameters反映,即模型的参数量. ...

tensorflow2计算flops

tensorflow2计算flops相关推荐

最新文章

热门文章