Skip to content
Snippets Groups Projects
  • Mashiro's avatar
    f04fec73
    [Feature]: add base model, ddp model wrapper and unit test (#268) · f04fec73
    Mashiro authored
    * add base model, ddp model and unit test
    
    * add unit test
    
    * fix unit test
    
    * fix docstring
    
    * fix cpu unit test
    
    * refine base data preprocessor
    
    * refine base data preprocessor
    
    * refine interface of ddp module
    
    * remove optimizer hook
    
    * add forward
    
    * fix as comment
    
    * fix unit test
    
    * fix as comment
    
    * fix build optimizer wrapper
    
    * rebase main and fix unit test
    
    * stack_batch support stacking ndim tensor, add docstring for merge dict
    
    * fix lint
    
    * fix test loop
    
    * make precision_context effective to data_preprocessor
    
    * fix as comment
    
    * fix as comment
    
    * refine docstring
    
    * change collate_data output typehints
    
    * rename to_rgb to bgr_to_rgb and rgb_to_bgr
    
    * support build basemodel with built DataPreprocessor
    
    * fix as comment
    
    * fix docstring
    [Feature]: add base model, ddp model wrapper and unit test (#268)
    Mashiro authored
    * add base model, ddp model and unit test
    
    * add unit test
    
    * fix unit test
    
    * fix docstring
    
    * fix cpu unit test
    
    * refine base data preprocessor
    
    * refine base data preprocessor
    
    * refine interface of ddp module
    
    * remove optimizer hook
    
    * add forward
    
    * fix as comment
    
    * fix unit test
    
    * fix as comment
    
    * fix build optimizer wrapper
    
    * rebase main and fix unit test
    
    * stack_batch support stacking ndim tensor, add docstring for merge dict
    
    * fix lint
    
    * fix test loop
    
    * make precision_context effective to data_preprocessor
    
    * fix as comment
    
    * fix as comment
    
    * refine docstring
    
    * change collate_data output typehints
    
    * rename to_rgb to bgr_to_rgb and rgb_to_bgr
    
    * support build basemodel with built DataPreprocessor
    
    * fix as comment
    
    * fix docstring