Skip to content
Snippets Groups Projects
  1. Jun 13, 2022
  2. Jun 09, 2022
    • Mashiro's avatar
      [Enhance] Enhance img data preprocessor (#290) · 931db990
      Mashiro authored
      * fix BaseDataPreprocessor
      
      * fix BaseDataPreprocessor
      
      * change device type to torch.device
      
      * change device type to torch.device
      
      * fix cpu method of base model
      
      * Allow ImgDataPreprocessor do not normalize
      
      * remove unnecessary type ignore
      
      * make mean and std optional
      
      * refine docstring
      931db990
    • Mashiro's avatar
      [Fix] Fix BaseDataPreprocessor and BaseModel (#285) · a9afdad7
      Mashiro authored
      * fix BaseDataPreprocessor
      
      * fix BaseDataPreprocessor
      
      * change device type to torch.device
      
      * change device type to torch.device
      
      * fix cpu method of base model
      a9afdad7
  3. Jun 07, 2022
    • Mashiro's avatar
      [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
      f04fec73
Loading