f64051041 / SARAS-Net

58 stars 9 forks source link

Why the count of parameters and flops differ from that shown in the paper? #9

Open Casey-bit opened 11 months ago

Casey-bit commented 11 months ago

image

Here are my codes.

model = Change_detection() input1 = torch.rand((1, 3, 256, 256)) input2 = torch.rand((1, 3, 256, 256)) from fvcore.nn import FlopCountAnalysis, flop_count_table model.eval() flops = FlopCountAnalysis(model, (input1, input2)) print(flop_count_table(flops))

image

Thank you for help.