/keras-flops

FLOPs calculator with tf.profiler for neural network architecture written in tensorflow 2.2+ (tf.keras)

Primary LanguagePythonMIT LicenseMIT

Stargazers