Skip to content

Variable input resolutions for ResNets #140

Closed
@macaodha

Description

@macaodha

The current implementation of the ResNet models has a fixed 7x7 average pooling layer before the final FC layer.

If this was changed to nn.AdaptiveAvgPool2d(1) it would allow for variable sized input batches (e.g. like the 320x320 batches used at test time in the 'Identity Mappings in Deep Residual Network' paper)? Although I'm not sure if this will incur a performance hit?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions