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

ResUNet doesn't match paper's parameters #3

Open
jchen42703 opened this issue Dec 19, 2019 · 0 comments
Open

ResUNet doesn't match paper's parameters #3

jchen42703 opened this issue Dec 19, 2019 · 0 comments
Labels
bug Something isn't working

Comments

@jchen42703
Copy link
Owner

Author said it's supposed to only have 2.2M parameters. This implementation's ResUNet has 2.4M with:

    model = ResUNet(input_channels=5, base_num_features=16, num_classes=3,
                    num_pool=4, max_num_features=256)
    # calculating # of parameters
    total = sum(p.numel() for p in model.parameters())
    trainable = sum(p.numel() for p in model.parameters() if p.requires_grad)
    print(f"Total # of Params: {total}\nTrainable params: {trainable}")

Not sure why. Probably doesn't have to do with the biases. Might be due to the parameters in 1 single extra expensive conv.
#2

@jchen42703 jchen42703 added the bug Something isn't working label Dec 19, 2019
@jchen42703 jchen42703 pinned this issue Dec 22, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

1 participant