1. 07 Jul, 2022 1 commit
    • Zaida Zhou's avatar
      [Feature] Add support for mps (#2092) · 6a03918f
      Zaida Zhou authored
      * [Feature] Add support for MPS
      
      * fix import error
      
      * update ut
      
      * fix error
      
      * trigger CI
      
      * use a unique basename for test file modules
      
      * avoid bc-breaking
      6a03918f
  2. 15 Apr, 2022 1 commit
    • Qiza-lyhm's avatar
      [Fix] Fix the output of scatter on MLU device (#1881) · 95273020
      Qiza-lyhm authored
      To have the same shape of MLU copy and Scatter.forward, a 1-dim list has to be added.
      
      Origin MMDP / MMDDP with this scatter_gather returns a list of tensor for each targets, and here MLU copy returns the same shape Tensor in a list of just one default target.
      95273020
  3. 14 Apr, 2022 1 commit
  4. 08 Aug, 2021 1 commit
  5. 19 Aug, 2020 1 commit
    • Wenwei Zhang's avatar
      Fix mmcv-dataparallel (#497) · 7a6285b1
      Wenwei Zhang authored
      * Fix mmcv-dataparallel
      
      * Fix (parallel): fix CPU inference with MMDataParallel
      
      * Update docstrings
      
      * Doc (parallel): refine docstrings
      
      * Fix (parallel): fix missing changes of train/val step function
      
      * resolve comments
      
      * Fix (data_parallel): fix bug when single gpu test return None
      7a6285b1
  6. 03 Jul, 2020 1 commit
  7. 10 Jan, 2020 1 commit
  8. 08 Oct, 2018 1 commit