Skip to content
Snippets Groups Projects
  • Mashiro's avatar
    a9b6753f
    Make TTAModel compatible with FSDP (#611) · a9b6753f
    Mashiro authored
    
    * Add build_runner_with_tta and PrepareTTAHook
    
    * rename hook file
    
    * support build tta runner with runner type
    
    * add unit test
    
    * Add build_runner_with_tta to index.rst
    
    * minor refine
    
    * Add runner test cast
    
    * Fix unit test
    
    * fix unit test
    
    * tmp save
    
    * pop None if key does not exist
    
    * Fix is_model_wrapper and force register class in test_runner
    
    * [Fix] Fix is_model_wrapper
    
    * destroy group after ut
    
    * register module in testcase
    
    * pass through unit test
    
    * fix as comment
    
    * remove breakpoint
    
    * remove mmengine/testing/runner_test_cast.py
    
    * minor refine
    
    * minor refine
    
    * minor refine
    
    * set default data preprocessor for model
    
    * minor refine
    
    * minor refine
    
    Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
    
    * fix lint
    
    * Fix unit test
    
    * replace  with  in ImgDataPreprocessor
    
    * Fix as comment
    
    * add inference tutorial in advanced tutorial
    
    * update index.rst
    
    * add tta example
    
    * refine tta tutorial
    
    * Add english tutorial
    
    * add note for build_runner_with_tta
    
    * Fix as comment
    
    * add examples
    
    * remove chinese comment
    
    * Update docs/en/advanced_tutorials/test_time_augmentation.md
    
    Co-authored-by: default avatarRangiLyu <lyuchqi@gmail.com>
    
    Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
    Co-authored-by: default avatarRangiLyu <lyuchqi@gmail.com>
    Make TTAModel compatible with FSDP (#611)
    Mashiro authored
    
    * Add build_runner_with_tta and PrepareTTAHook
    
    * rename hook file
    
    * support build tta runner with runner type
    
    * add unit test
    
    * Add build_runner_with_tta to index.rst
    
    * minor refine
    
    * Add runner test cast
    
    * Fix unit test
    
    * fix unit test
    
    * tmp save
    
    * pop None if key does not exist
    
    * Fix is_model_wrapper and force register class in test_runner
    
    * [Fix] Fix is_model_wrapper
    
    * destroy group after ut
    
    * register module in testcase
    
    * pass through unit test
    
    * fix as comment
    
    * remove breakpoint
    
    * remove mmengine/testing/runner_test_cast.py
    
    * minor refine
    
    * minor refine
    
    * minor refine
    
    * set default data preprocessor for model
    
    * minor refine
    
    * minor refine
    
    Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
    
    * fix lint
    
    * Fix unit test
    
    * replace  with  in ImgDataPreprocessor
    
    * Fix as comment
    
    * add inference tutorial in advanced tutorial
    
    * update index.rst
    
    * add tta example
    
    * refine tta tutorial
    
    * Add english tutorial
    
    * add note for build_runner_with_tta
    
    * Fix as comment
    
    * add examples
    
    * remove chinese comment
    
    * Update docs/en/advanced_tutorials/test_time_augmentation.md
    
    Co-authored-by: default avatarRangiLyu <lyuchqi@gmail.com>
    
    Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
    Co-authored-by: default avatarRangiLyu <lyuchqi@gmail.com>