- 博客(1)
- 资源 (4)
- 收藏
- 关注
原创 tf.keras计算FLOPs
tf.keras计算FLOPs推荐一个好用的包可以通过pip安装pip install keras_flopsimport tensorflow as tffrom keras_flops import get_flopsimport pathlibimport keras.backend as K# import osimport numpy as npMnasnet = tf.keras.Sequential([ tf.keras.layers.InputLayer(in
2021-04-15 16:51:16
3631
9
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人