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

Encoder APIs: Features #175

Open
FrancescoSaverioZuppichini opened this issue Dec 24, 2020 · 0 comments
Open

Encoder APIs: Features #175

FrancescoSaverioZuppichini opened this issue Dec 24, 2020 · 0 comments
Labels
enhancement New feature or request

Comments

@FrancescoSaverioZuppichini
Copy link
Owner

Would be nice to also pass the stages + features_widths parameters to a Encoder so we can on-the-fly change which layer is used.

ResNetEncoder(32, 64, stages=lambda enc: [enc.layers[0]], enc.layers[1]] , features_widths=lambda enc: [enc.start_features, enc.widths[0])

With the current API we could subclass the encoder

MyResNetEncoder(ResNetDecoder):

    @property
    def stages():
        return [self.stem[0],
                self.layers[1]]

    @property
    def features_widths():
        return [self.start_features[0],
                self.widths[0]]
@FrancescoSaverioZuppichini FrancescoSaverioZuppichini added the enhancement New feature or request label Dec 24, 2020
@FrancescoSaverioZuppichini FrancescoSaverioZuppichini changed the title Features Encoder APIs: Features Dec 25, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant