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

filter parameter not used for blocks.vgg_fc #27

Open
NickleDave opened this issue Sep 2, 2017 · 1 comment
Open

filter parameter not used for blocks.vgg_fc #27

NickleDave opened this issue Sep 2, 2017 · 1 comment

Comments

@NickleDave
Copy link

Currently block.vgg_fc has a filters argument but the calls to set up the layers are set to a constant 4096

def vgg_fc(filters, weight_decay=0., block_name='block5'):
...
    def f(x):
        fc6 = Conv2D(filters=4096, kernel_size=(7, 7),
                     activation='relu', padding='same',
                     dilation_rate=(2, 2),
                     kernel_initializer='he_normal',
                     kernel_regularizer=l2(weight_decay),
name='{}_fc6'.format(block_name))(x)
...

Is that intentional? For my use case I needed to decrease the number of filters.

I'd be happy to submit a PR if that's just a minor bug fix

@JihongJu
Copy link
Owner

JihongJu commented Oct 10, 2017

Hi. You are more than welcome to send a PR. Actually, all the arguments for the Conv layers, or any other layers that have arguments, should be changeable.

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