• Vasilis Vryniotis's avatar
    Add registration mechanism for models (#6333) · 0a919dbb
    Vasilis Vryniotis authored
    * Model registration mechanism.
    
    * Add overwrite options to the dataset prototype registration mechanism.
    
    * Adding example models.
    
    * Fix module filtering
    
    * Fix linter
    
    * Fix docs
    
    * Make name optional if same as model builder
    
    * Apply updates from code-review.
    
    * fix minor bug
    
    * Adding getter for model weight enum
    
    * Support both strings and callables on get_model_weight.
    
    * linter fixes
    
    * Fixing mypy.
    
    * Renaming `get_model_weight` to `get_model_weights`
    
    * Registering all classification models.
    
    * Registering all video models.
    
    * Registering all detection models.
    
    * Registering all optical flow models.
    
    * Fixing mypy.
    
    * Registering all segmentation models.
    
    * Registering all quantization models.
    
    * Fixing linter
    
    * Registering all prototype depth perception models.
    
    * Adding tests and updating existing tests.
    
    * Fix linters
    
    * Fix tests.
    
    * Add beta annotation on docs.
    
    * Fix tests.
    
    * Apply changes from code-review.
    
    * Adding documentation.
    
    * Fix docs.
    0a919dbb
resnet.py 17.4 KB