Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Params in table2 #26

Open
Pwoer-zy opened this issue May 29, 2023 · 0 comments
Open

Params in table2 #26

Pwoer-zy opened this issue May 29, 2023 · 0 comments

Comments

@Pwoer-zy
Copy link

Pwoer-zy commented May 29, 2023

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!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant