weixin_41076537的博客

原创 tf.keras计算FLOPs

tf.keras计算FLOPs 推荐一个好用的包 可以通过pip安装 pip install keras_flops import tensorflow as tf from keras_flops import get_flops import pathlib import keras.backend as K # import os import numpy as np Mnasnet = tf.keras.Sequential([ tf.keras.layers.InputLayer(in