Open Pwoer-zy opened 1 year ago
Hi, May I ask how Params are calculated in Table 2 ? I used the following code to calculate Params, and the result obtained is significantly different from Table 2's 8.53 million. The result I got was 4.33 million.
import torch from thop import profile from src.anchor_based.dsnet import DSNet model = DSNet('attention', 1024, 128, [4, 8, 16, 32], 8) input = torch.randn(1, 1024, 1024) flops, params = profile(model, inputs=(input, )) print('flops:{}'.format(flops)) print('params:{}'.format(params))
I would like to know how you calculated it and look forward to receiving your reply. Thank you very much!
Hi, May I ask how Params are calculated in Table 2 ? I used the following code to calculate Params, and the result obtained is significantly different from Table 2's 8.53 million. The result I got was 4.33 million.
import torch from thop import profile from src.anchor_based.dsnet import DSNet model = DSNet('attention', 1024, 128, [4, 8, 16, 32], 8) input = torch.randn(1, 1024, 1024) flops, params = profile(model, inputs=(input, )) print('flops:{}'.format(flops)) print('params:{}'.format(params))
I would like to know how you calculated it and look forward to receiving your reply. Thank you very much!