"megatron/legacy/data/vit_dataset.py" did not exist on "e1f9c3a5e3e6f266cf0968619773c73b83954e8c"
  • Charlie Lin's avatar
    Dynamic ref pooling (#1449) · 0e40ebaa
    Charlie Lin authored
    Extends the pooling operators for dynamic shape inputs
    
    AveragePooling
    GlobalAveragePooling
    MaxPooling
    GlobalMaxPooling
    LpNormPooling
    GlobalLpNormPooling
    y.github.com>
    0e40ebaa