princeton-vl / RAFT

BSD 3-Clause "New" or "Revised" License
3.12k stars 621 forks source link

how to count FLOPS of RAFT model? #150

Open leeqiaogithub opened 1 year ago

leeqiaogithub commented 1 year ago

I wonder how to accurately count floating point operations per second of RAFT model. The open interface basically only counts the convolution , for those special operations in the model, are there any good statistical methods? In other words, how do you do that?