How to exclude flops of 1st input?

[Angiemaster]

I wanna exclude the t=1 for the total_ops & total_params, just need t=2~5, could you help me with it? thank you total_ops += m.total_ops total_params += m.total_params

My input is below, t = 5 A= torch.rand(1, t, 3, LR_h, LR_w).cuda() B= torch.rand(1, t, 3, FV_h, FV_w).cuda() macs, params = profile_origin(model, inputs=(A,B))

[ivanstepanovftw]

Ask here: chat.openai.com