Skip to content

Add nn::AdaptiveAvgPool2D #903

@quietlychris

Description

@quietlychris

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?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions