1. 29 Aug, 2021 1 commit
    • Eugene Liu's avatar
      Add DCN and Modulated DCN CPU implementation (#1278) · e621e08d
      Eugene Liu authored
      * DCN cpu version
      
      * add modulated dcn cpu version
      
      * move deform_conv_shape_check to deform conv utils
      
      * add inline to deform_conv_shape_check
      
      * add tests
      
      * run linter
      
      * add newline at file end
      
      * run pre-commit against modulated deform conv cpp
      
      * update saconv test
      
      * run clang-format
      
      * remove cuda device inline
      
      * refactor dcn cuda/cpu functions
      
      * remove DCN util
      
      * remove DCN util hpp from all included files
      
      * Addressing PR comment by refactoring modulated-DCN
      
      * fix lint in cpp files
      e621e08d
  2. 24 Aug, 2021 1 commit
    • Wenwei Zhang's avatar
      [Feature] Add windows CI (#1023) · 979a355d
      Wenwei Zhang authored
      
      
      * add windows CI
      
      * clean versions
      
      * only allow pt1.7 on windows
      
      * fix windows install issue
      
      * add win cpu
      
      * fix win command
      
      * clean unnecessary command
      
      * resolve turbojpeg & tempfile on win
      
      * replace os.readlink with os.path.realpath
      
      * fix windows ci
      
      * close file before removing it
      
      * fix windows ci
      
      * fix symlink on windows
      
      * fix windows ci
      
      * fix windows ci
      
      * fix windows ci
      
      * fix windows ci
      
      * fix windows ci
      
      * fix windows ci
      
      * fix windows ci
      
      * fix windows ci
      
      * fix windows ci
      
      * modify according to comment
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      979a355d
  3. 23 Aug, 2021 3 commits
    • Paige Wang's avatar
      add modulated_deform_conv in onnxruntime support (#1281) · 1790e9f2
      Paige Wang authored
      * add modulated_deform_conv in onnxruntime support
      
      * Add docs descriptions
      
      * Add gpu test in test_onnx.py
      
      * code format
      
      * remove new usage and move if outside for loop
      
      * use memset when bias is nullptr
      1790e9f2
    • Ma Zerun's avatar
      [WIP] continue PR #784 (#1221) · 357b0dfb
      Ma Zerun authored
      
      
      * Add gradient cumulative optimizer
      
      fixes #190
      
      * Update optimizer.py
      
      * Update optimizer.py
      
      * fix loss scale improperly in last equivalent_iter
      
      * Add `GradientCumulativeOptimizerHook` in `__init__.py`.
      
      * Add docstring of `GradientCumulativeOptimizerHook`.
      
      * Add type check, BN warning and resume warning. And fix typo, lint the
      code.
      
      * Add unit test
      
      * Update docstring example.
      
      * Change GradientCumulativeOptimizerHook `__init__` arguments.
      
      * Add GradientCumulativeOptimzierHook unit tests with IterBasedRunner.
      
      * Add GradientCumulativeFp16OptimizerHook.
      
      * Add unit tests of GradientCumulativeFp16OptimizerHook
      
      * Use '!=' instead of '>' to determine resume
      Co-authored-by: default avatarZhiyuan Chen <this@zyc.ai>
      357b0dfb
    • Yining Li's avatar
      Support deprecation info in Config (#1275) · 4d9b43de
      Yining Li authored
      * Support deprecation checking in Config
      
      * add unittest for config deprecation
      
      * support reference link in deprecation info
      
      * add doc
      
      * Update config.md
      4d9b43de
  4. 11 Aug, 2021 2 commits
    • Zaida Zhou's avatar
      [Fix] Fix the bug that the training log and evaluating log are mixed (#1252) · 846d3a4a
      Zaida Zhou authored
      * [Fix] Fix the bug that training log and evaluating log are mixed
      
      * [Fix] Fix the bug that training log and evaluating log are mixed
      
      * fix comment
      
      * fix import error
      
      * refactor
      
      * refactor
      
      * refactor
      
      * clear log_buffer before evaluation
      
      * fix error
      
      * add unittest
      846d3a4a
    • Shilong Zhang's avatar
      Fix potential bugs of basemodule when record the initilization information (#1217) · 18c64d5f
      Shilong Zhang authored
      * add logger for init
      
      * change init_info of oevrload init_weight
      
      * add judgement for params_init_info
      
      * add delete comments for params_init_info
      
      * add docstr and more comments
      
      * add docstr and more comments
      
      * resolve comments
      
      * dump to a file
      
      * add unitest
      
      * fix unitest
      
      * fix unitest
      
      * write to ori log
      
      * fix typo
      
      * resolve commnets
      
      * fix call initweights twice in topmost module
      
      * fix the potential bug of recursive import
      
      * fix unitest
      
      * fix potiential bugs
      
      * remove unneccesary change
      
      * add more unitest
      
      * fix add param in initweights
      
      * add more comments
      
      * raise error
      
      * add more detail assert error
      18c64d5f
  5. 08 Aug, 2021 1 commit
  6. 23 Jul, 2021 2 commits
    • Shilong Zhang's avatar
      [Features] Add logger for initialization of parameters (#1150) · 17fa6670
      Shilong Zhang authored
      * add logger for init
      
      * change init_info of oevrload init_weight
      
      * add judgement for params_init_info
      
      * add delete comments for params_init_info
      
      * add docstr and more comments
      
      * add docstr and more comments
      
      * resolve comments
      
      * dump to a file
      
      * add unitest
      
      * fix unitest
      
      * fix unitest
      
      * write to ori log
      
      * fix typo
      
      * resolve commnets
      17fa6670
    • 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
  7. 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
  8. 14 Jul, 2021 1 commit
  9. 13 Jul, 2021 1 commit
  10. 09 Jul, 2021 2 commits
  11. 03 Jul, 2021 1 commit
  12. 29 Jun, 2021 2 commits
  13. 25 Jun, 2021 4 commits
  14. 24 Jun, 2021 3 commits
  15. 23 Jun, 2021 1 commit
  16. 16 Jun, 2021 1 commit
  17. 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
  18. 08 Jun, 2021 1 commit
  19. 01 Jun, 2021 1 commit
  20. 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
  21. 25 May, 2021 3 commits
  22. 24 May, 2021 1 commit
  23. 23 May, 2021 3 commits