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