-
-
Notifications
You must be signed in to change notification settings - Fork 98
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
Add nn::AdaptiveAvgPool2D
#903
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Re-opening per #215:
I was working on implementing AlexNet in dfdx per https://github.com/LaurentMazare/tch-rs/blob/main/src/vision/alexnet.rs, but have gotten a bit blocked by not having this layer type available. Would it be possible to add this, or alternatively to have some documentation on constructing the equivalent using
GlobalAvgPool
that was mentioned in that issue?The text was updated successfully, but these errors were encountered: