1. 23 Jul, 2021 1 commit
    • Haodong Duan's avatar
      [Improvement] Improve digit_version & use it for version_checking (#1185) · ef48a473
      Haodong Duan authored
      * improve digit_version & use it for version_checking
      
      * more testing for digit_version
      
      * setuptools >= 50 is needed
      
      * fix CI
      
      * add debuging log
      
      * >= to ==
      
      * fix lint
      
      * remove
      
      * add failure case
      
      * replace
      
      * fix
      
      * consider TORCH_VERSION == 'parrots'
      
      * add unittest
      
      * digit_version do not deal with the case if 'parrots' in version name.
      ef48a473
  2. 20 Jul, 2021 2 commits
    • Gu Wang's avatar
      [Feature] Add flat cosine lr updater (#1066) · 025ee9f5
      Gu Wang authored
      * add flat cosine lr updater
      
      * add test
      
      * add doc
      
      * update doc
      
      * reformat
      
      * update unittest
      
      * update test flat cos
      
      * remove momentum hook test
      
      * update test
      
      * change assert to ValueError
      
      * fix unittest
      
      * add by_epoch=True unittest
      
      * change to start_percent
      
      * change to start_percent in test
      025ee9f5
    • q.yao's avatar
      [Feature] Add ONNX export support to torch.roll (#1194) · 5f9e6b61
      q.yao authored
      * add torch.roll to onnx
      
      * remove skip test
      
      * add support to torch<170
      
      * add dim=0 for torch==1.9.0
      
      * update fixture function name, add comment in roll symbolic
      5f9e6b61
  3. 14 Jul, 2021 1 commit
  4. 13 Jul, 2021 1 commit
  5. 09 Jul, 2021 2 commits
  6. 03 Jul, 2021 1 commit
  7. 29 Jun, 2021 2 commits
  8. 25 Jun, 2021 4 commits
  9. 24 Jun, 2021 3 commits
  10. 23 Jun, 2021 1 commit
  11. 16 Jun, 2021 1 commit
  12. 11 Jun, 2021 2 commits
    • Shilong Zhang's avatar
      Refactor the baseclass related to transformer (#978) · e05fb560
      Shilong Zhang authored
      
      
      * minor changes
      
      * change to modulist
      
      * change to Sequential
      
      * replace dropout with attn_drop and proj_drop in MultiheadAttention
      
      * add operation_name for attn
      
      * add drop path and move all ffn args to ffncfgs
      
      * fix typo
      
      * fix a bug when use default value of ffn_cfgs
      
      * fix ffns
      
      * add deprecate warning
      
      * fix deprecate warning
      
      * change to pop kwargs
      
      * support register FFN of transformer
      
      * support batch first
      
      * fix batch first wapper
      
      * fix forward wapper
      
      * fix typo
      
      * fix lint
      
      * add unitest for transformer
      
      * fix unitest
      
      * fix equal
      
      * use allclose
      
      * fix comments
      
      * fix comments
      
      * change configdict to dict
      
      * move drop to a file
      
      * add comments for drop path
      
      * add noqa 501
      
      * move bnc wapper to MultiheadAttention
      
      * move bnc wapper to MultiheadAttention
      
      * use dep warning
      
      * resolve comments
      
      * add unitest:
      
      * rename residual to identity
      
      * revert runner
      
      * msda residual to identity
      
      * rename inp_identity to identity
      
      * fix name
      
      * fix transformer
      
      * remove key in msda
      
      * remove assert for key
      Co-authored-by: default avatarHIT-cwh <2892770585@qq.com>
      Co-authored-by: default avatarbkhuang <congee524@gmail.com>
      Co-authored-by: default avatarWenwei Zhang <40779233+ZwwWayne@users.noreply.github.com>
      e05fb560
    • Dmitry Sidnev's avatar
      [Feature] enable exporting to onnx for PointRend (#953) · a88d1d28
      Dmitry Sidnev authored
      
      
      * Fix export to onnx for PointRend
      
      * Fix codestyle
      
      * Fix codestyle
      
      * Fix type in docstring
      
      * Minor fix
      
      * Fix export with custom ops
      
      * Fix codestyle
      
      * Add tests for bilinear_grid_sample function
      
      * Remove redundant operation and rename variables
      
      * Fix bug in bilinear_grid_sample and update test
      
      * Fix getting batch size
      
      * skip torch==1.3.1
      
      * remove unused import
      
      * fix lint
      
      * support export with batch
      
      * fix dynamic clip
      
      * skip test for torch<1.5.0
      
      * Add docstrings and comments
      
      * Minor fix
      
      * Recover clipping code
      
      * Fix clamping in pytorch 1.7.0
      
      * Fix bilinear_grid_sampler
      
      * Minor fix
      Co-authored-by: default avatarmaningsheng <maningsheng@sensetime.com>
      a88d1d28
  13. 08 Jun, 2021 1 commit
  14. 01 Jun, 2021 1 commit
  15. 31 May, 2021 1 commit
    • SemyonBevzuk's avatar
      [Feature] NMS update (#957) · bf2c9fa8
      SemyonBevzuk authored
      * Add score_threshold and max_num to NMS
      
      * Fix codestyle
      
      * Fix codestyle
      
      * Fix inds in nms
      
      * Update nms docstring
      
      * Move score_threshold and max_num arguments
      
      * Fix args order in docstring
      
      * fix lint of c++ file
      
      * Remove torch.onnx.is_in_onnx_export() and add max_num to batched_nms for separate classes.
      
      * Rewrote max_num handling in NMSop.symbolic
      
      * Added processing max_output_boxes_per_class when exporting to TensorRT
      
      * Added score_threshold and max_num for NMS in test_onnx.py and test_tensorrt.py
      
      * Remove _is_value(max_num)
      
      * fix ci errors with torch==1.3.1
      
      * Update test_batched_nms in test_nms.py
      
      * Added tests for preprocess_onnx
      
      * Moved 'test_tensorrt_preprocess.py' and 'preprocess', updated 'remove_tmp_file'.
      
      * Update mmcv/tensorrt/__init__.py
      
      * Fix segfault torch==1.3.1 (remove onnx.checker.check_model)
      
      * Returned 'onnx.checker.check_model' with torch version check
      
      * Changed torch version from 1.3.1 to 1.4.0
      
      *...
      bf2c9fa8
  16. 25 May, 2021 3 commits
  17. 24 May, 2021 1 commit
  18. 23 May, 2021 4 commits
  19. 13 May, 2021 3 commits
  20. 12 May, 2021 1 commit
    • jeffreykuang's avatar
      Add pixel group and contour expand ops (#993) · 2623fbf2
      jeffreykuang authored
      * add pixel group ops
      
      * reformatting
      
      * formatting&rm auto
      
      * Add citation
      
      * Add contour expand
      
      * c++ linting
      
      * Add unit tests with Tensor
      
      * rm model.pth
      
      * rename
      
      * c++ linting
      
      * c++ linting
      
      * Rename variables
      2623fbf2
  21. 11 May, 2021 2 commits
  22. 10 May, 2021 2 commits