Skip to content
Snippets Groups Projects
  1. Jun 28, 2022
  2. Jun 21, 2022
    • Alex Yang's avatar
      [Feat] Support FSDP Training (#304) · ef946404
      Alex Yang authored
      * [Feat] Support FSDP Training
      
      * fix version comparison
      
      * change param format and move `FSDP_WRAP_POLICY` to wrapper file
      
      * add docstring and type hint,reformat code
      
      * fix type hint
      
      * fix typo, reformat code
  3. Jun 13, 2022
  4. Jun 09, 2022
  5. 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
  6. May 19, 2022
    • RangiLyu's avatar
      [Feature] Support EMA and SWA. (#239) · 0279ac2e
      RangiLyu authored
      * [Feature] Support EMA and SWA.
      
      * add ema hook
      
      * add avg model ut
      
      * add more unit tests
      
      * resolve comments
      
      * fix warmup ema
      
      * rename
      
      * fix comments
      
      * add assert
      
      * fix typehint
      
      * add comments
  7. Feb 26, 2022
Loading