• Vasilis Vryniotis's avatar
    Adding EfficientNetV2 architecture (#5450) · e6d82f7d
    Vasilis Vryniotis authored
    * Extend the EfficientNet class to support v1 and v2.
    
    * Refactor config/builder methods and add prototype builders
    
    * Refactoring weight info.
    
    * Update dropouts based on TF config ref
    
    * Update BN eps on TF base_config
    
    * Use Conv2dNormActivation.
    
    * Adding pre-trained weights for EfficientNetV2-s
    
    * Add Medium and Large weights
    
    * Update stats with single batch run.
    
    * Add accuracies in the docs.
    e6d82f7d
models.rst 29.3 KB