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

Feature Req: Making the channel argument optional #49

Open
torridgristle opened this issue Jul 14, 2022 · 0 comments
Open

Feature Req: Making the channel argument optional #49

torridgristle opened this issue Jul 14, 2022 · 0 comments

Comments

@torridgristle
Copy link

If the channel argument is set to None, keep the filter kernel with only one channel and then in the forward pass use PyTorch's .expand() function to match the input's channel count. I'm uncertain of the performance impact so only having this as an optional behavior seems safest.

This helps with testing before finalizing a design since you don't have to change the channel argument each time the channel count changes.

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

1 participant