返回信息流代码是网上找的，只能在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么？
万分感谢！
这是一条镜像帖。来源：北邮人论坛 / python / #26138同步于 2022/4/21
该镜像源已超过 30 天没有更新，可能在源站已被删除。
Python机器人发帖
【求助】tensorflow2.1计算模型flops
jingyutui
2022/4/21镜像同步4 回复
订阅后，新回复会通过你的通知中心匿名送达。
4 条回复
都是开源的，实在不行抄代码移植？
【 在 jingyutui 的大作中提到: 】
: 代码是网上找的，只能在tensorflow2.0及2.1下运行，现在需要计算模型的flops
: 尝试过安装包keras_flops，无奈该包只能在2.2+环境下运行
: ............
代码移植对我有点难[ema1]，本人菜鸡一枚，之前想尝试把代码改成2.2+版本可用的，搞了半天一个报错都没解决
【 在 paopjian 的大作中提到: 】
: 都是开源的，实在不行抄代码移植？
试试在GitHub上搜索一下模型，有可能有新的呢
【 在 jingyutui 的大作中提到: 】
: 代码移植对我有点难[ema1]，本人菜鸡一枚，之前想尝试把代码改成2.2+版本可用的，搞了半天一个报错都没解决