1. 11 Nov, 2022 1 commit
    • toni057's avatar
      Adding FLOPs and size to model metadata (#6936) · deba0562
      toni057 authored
      
      
      * Adding FLOPs and size to model metadata
      
      * Adding weight size to quantization models
      
      * Small refactor of rich metadata
      
      * Removing unused code
      
      * Fixing wrong entries
      
      * Adding .DS_Store to gitignore
      
      * Renaming _flops to _ops
      
      * Adding number of operations to quantization models
      
      * Reflecting _flops change to _ops
      
      * Renamed ops and weight size in individual model doc pages
      
      * Linter fixes
      
      * Rounding ops to first decimal
      
      * Rounding num ops and sizes to 3 decimals
      
      * Change naming of columns.
      
      * Update tables
      Co-authored-by: default avatarToni Blaslov <tblaslov@fb.com>
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      deba0562
  2. 23 Sep, 2022 1 commit
    • Ponku's avatar
      MaxVit model (#6342) · 6b1646ca
      Ponku authored
      
      
      * Added maxvit architecture and tests
      
      * rebased + addresed comments
      
      * Revert "rebased + addresed comments"
      
      This reverts commit c5b28398cd48d2f3403c7c8eeefbaba9df05fcfe.
      
      * Re-added model changes after revert
      
      * aligned with partial original implementation
      
      * removed submitit script fixed lint
      
      * mypy fix for too many arguments
      
      * updated old tests
      
      * removed per batch lr scheduler and seed setting
      
      * removed ontap
      
      * added docs, validated weights
      
      * fixed test expect, moved shape assertions in the begging for torch.fx compatibility
      
      * mypy fix
      
      * lint fix
      
      * added legacy interface
      
      * added weight link
      
      * updated docs
      
      * Update references/classification/train.py
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      
      * Update torchvision/models/maxvit.py
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      
      * adressed comments
      
      * update ra_maginuted and augmix_severity default values
      
      * adressed some comments
      
      * remove input_channels parameter
      Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
      6b1646ca