• Vasilis Vryniotis's avatar
    Adding ConvNeXt architecture in prototype (#5197) · afda28ac
    Vasilis Vryniotis authored
    * Adding CNBlock and skeleton architecture
    
    * Completed implementation.
    
    * Adding model in prototypes.
    
    * Add test and minor refactor for JIT.
    
    * Fix mypy.
    
    * Fixing naming conventions.
    
    * Fixing tests.
    
    * Fix stochastic depth percentages.
    
    * Adding stochastic depth to tiny variant.
    
    * Minor refactoring and adding comments.
    
    * Adding weights.
    
    * Update default weights.
    
    * Fix transforms issue
    
    * Move convnext to prototype.
    
    * linter fix
    
    * fix docs
    
    * Addressing code review comments.
    afda28ac
models.rst 27.7 KB