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

Total number of parameters of GA-Net-11 and GA-Net-15 #91

Open
motionlife opened this issue Aug 12, 2020 · 1 comment
Open

Total number of parameters of GA-Net-11 and GA-Net-15 #91

motionlife opened this issue Aug 12, 2020 · 1 comment

Comments

@motionlife
Copy link

The total number of GC-Net is 2.9 million; (2850785 according to my TensorFlow implementation)
The total number of PSM-Net is 5.2 million; (5224768 according to my TensorFlow implementation and original author's code)

Can anyone provide me the total number of parameters of GA-Net-11 and GA-Net-15? The author doesn't seem to be consistent when reporting those numbers in the paper. They omitted the whole feature extraction layers' parameters for PSM-net and their own model. However, I found that actually there feature extraction layers are crucial to the final results, more crucial than those aggregation net in terms of final performance. So I think it's important to compare the total number of model parameters.

@18651626101
Copy link

18651626101 commented Dec 16, 2021

The total number of GC-Net is 2.9 million; (2850785 according to my TensorFlow implementation) The total number of PSM-Net is 5.2 million; (5224768 according to my TensorFlow implementation and original author's code)

Can anyone provide me the total number of parameters of GA-Net-11 and GA-Net-15? The author doesn't seem to be consistent when reporting those numbers in the paper. They omitted the whole feature extraction layers' parameters for PSM-net and their own model. However, I found that actually there feature extraction layers are crucial to the final results, more crucial than those aggregation net in terms of final performance. So I think it's important to compare the total number of model parameters.

Hi, I have the same question as yours. I find that the total number of PSMNet is 5.2 million in Jia-Ren's code, however the total number of PSMNet is 3.5 million mentioned in GA-Net. Besides, when I calculate the total number of GA-Net-15, I get the different result compared with 2.3 million in the paper. I re-read the paper and I think you are right. The parameter mentioned in the paper is maybe just the 3Dconv part, ignoring the feature extraction layers.

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

2 participants