Releases: tokusumi/keras-flops
Releases · tokusumi/keras-flops
0.1.2: Add/add v2 op (#11)
Upload to PyPi
0.1.1 Upload to PyPi (#10)
0.1.0
first realese
- FLOPs calculator with tf.profiler for tf.keras (v2.2+)
- Add some tf.python.framework.ops.RegisterStatistics to supports some tf.keras.layers (only for GloblMaxPooling and BatchNormalization)
- Test some layers to make sure that FLOPs can be calculated correctly or not