flops-counter.pytorch icon indicating copy to clipboard operation
flops-counter.pytorch copied to clipboard

Published 20 hours ago • verified publisher icon sovrasov

FLOPS for LSTM model

Open preksha12 opened this issue 5 years ago • 2 comments

How to compute FLOPS for LSTM model for skeleton data (NTU-RGBD)? Error is due to input_res parameter as for skeleton data what should be this value.

preksha12 avatar Aug 15 '20 05:08 preksha12

Hi! I don't know what shape is exactly suitable for your model. If it has two inputs, you can arrange the inputs arguments in a custom way, like in #14.

sovrasov avatar Aug 17 '20 12:08 sovrasov

Actually I am trying to calculate FLOPS for NTU-RGBD dataset with skeleton data.

preksha12 avatar Sep 17 '20 07:09 preksha12