-
Notifications
You must be signed in to change notification settings - Fork 50
Open
Description
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!
Metadata
Metadata
Assignees
Labels
No labels