计算参数量和FLOPS的问题

[Krystal9999]

请问你是如何计算RWKV部分的Params和FLOPS的呢，有具体的测试代码吗

[Yaziwel]

请问你是如何计算RWKV部分的Params和FLOPS的呢，有具体的测试代码吗

你好，Restore_RWKV.py代码末尾就说计算参数和flops的代码，不过omnishift那里需要手动注释掉其他branch，只保留5x5卷积的参数。