返回信息流代码是网上找的，只能在tensorflow2.0及2.1下运行，现在需要计算模型的flops
尝试过安装包keras_flops，无奈该包只能在2.2+环境下运行
尝试用profile函数，无奈该函数只能在1.x环境下运行
从github上找到了一个model_profiler的包，2.1环境下测试例可以正常运行，但是换成自己的模型就报错
Traceback (most recent call last):
File "C:\Users\Desktop\csi\modelTrain.py", line 69, in <module>
profile = model_profiler(encoder, BATCH_SIZE)
File "D:\Study_Softwares\Anaconda3\envs\tensorflow\lib\site-packages\model_profiler\profiler.py", line 40, in model_profiler
mem = keras_model_memory_usage(use_units[2], model, Batch_size)
File "D:\Study_Softwares\Anaconda3\envs\tensorflow\lib\site-packages\model_profiler\utils.py", line 22, in deco
return multiplier.get(units, -1) * fn(*args, **kwds)
File "D:\Study_Softwares\Anaconda3\envs\tensorflow\lib\site-packages\model_profiler\utils.py", line 81, in keras_model_memory_usage
single_layer_mem *= s
TypeError: can't multiply sequence by non-int of type 'tuple'
在网上看了一堆都没解决，求问各位大佬有解决办法么？或者有别的方法可以在2.1环境下计算出模型flops么？
万分感谢！
这是一条镜像帖。来源：北邮人论坛 / ml-dm / #38361同步于 2022/4/21
ML_DM机器人发帖
【求助】tensorflow2.1环境下计算模型flops
jingyutui
2022/4/21镜像同步0 回复
订阅后，新回复会通过你的通知中心匿名送达。
0 条回复
暂无回复 · 你可以订阅本帖等待新回复。