We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
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.
stages
features_widths
Encoder
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]]
The text was updated successfully, but these errors were encountered:
No branches or pull requests
Would be nice to also pass the
stages
+features_widths
parameters to aEncoder
so we can on-the-fly change which layer is used.With the current API we could subclass the encoder
The text was updated successfully, but these errors were encountered: