liuyuan1-v
0d1b224fb1
[Feature] Support NmsRotated with cambricon MLU backend ( #2643 )
...
* [Feature] Support NmsRotated with cambricon MLU backend
* [Feature] remove foolproofs in nms_rotated_mlu.cpp
* [Feature] fix lint in test_nms_rotated.py
* [Feature] fix kMLU not found in nms_rotated.cpp
* [Feature] modify mlu support in nms.py
* [Feature] modify nms_rotated support in ops.md
* [Feature] modify ops/nms.py
2023-03-23 11:36:32 +08:00
duzekun
06fa32853b
[Feature] Add MLU support for Sparse Convolution op ( #2589 )
...
* [Feature] Add sparse convolution MLU API
* [Feature] update cpp code style
* end-of-file
* delete libext.a
* code style
* update ops.md
---------
Co-authored-by: budefei <budefei@cambricon.com>
2023-03-20 19:28:13 +08:00
sherie
01a0f53ea4
[Feature] Add Ascend support for RoIAlign op ( #2638 )
...
* add roi_algin op
* add unit test
* update ops.md
* fix typo
---------
Co-authored-by: momo609 <963372609.qq.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-20 19:10:42 +08:00
liuhw
745985937a
[Feature] Add the support of voxelization op for ascend device ( #2613 )
...
* update
* update
2023-03-03 20:07:36 +08:00
Hakjin Lee
8e8ab22686
[Enhancement] Support MultiScaleDeformableAttention with AMP ( #2541 )
...
* [Enhance] Support FP16 for MSDeformAttn
* [Fix] Data type mismatch
* Update mmcv/ops/multi_scale_deform_attn.py
* Add UT
Author: nijkah <nijkah@gmail.com>
* Add cuda available condition
---------
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-02-17 19:27:14 +08:00
liuhw
5a61e534ea
[Feature] Add Ascend support for bbox_overlaps ( #2580 )
...
* Add support for Ascend devices with bbox_overlaps
* ...
* ...
2023-02-06 18:33:16 +08:00
xinliang123
615d28176a
[Feature] Add Ascend support for RoIPoolGrad op ( #2569 )
...
* add roipoolgrad adapter
* amend
2023-02-06 18:29:59 +08:00
enemy1205
422816e45c
[Feature] Add C++ implementation for bbox_overlaps ( #2477 )
...
* add ops bbox_overlaps
* format code
* Return the pytorch version
* Intermediate modification
* Solve problems in parameter passing
* revise bug
* "add test case"
2023-01-31 00:04:35 +08:00
ckirchhoff
54ed0ed869
[Feature] Add support for Ascend devices with nms_rotated ( #2550 )
...
* [Feature]: add nms_rotated npu adaptater code
* [BugFix]: modify param in nms_rotated_npu.cpp
* [clean code]: nms_rotated_npu.cpp
* [clean code]: nms_rotated_npu.cpp
* [clean code]: nms_rotated_npu.cpp
* [clean code]: nms_rotated.cpp
* [Doc]: add nms_rotated op in supported op list at ops.md
* [Test]: add nms_rotated unit_test
* [Bug]: remove device parameter in test_batched_nms function
2023-01-28 21:08:22 +08:00
liuduanhui
dfb03806a1
[Feature] Support ball_query with cambricon MLU backend and mlu-ops library. ( #2520 )
...
* [Feature] Support ball_query with cambricon MLU backend and mlu-ops library.
* [Fix] update operator data layout setting.
* [Fix] add cxx compile option to avoid symbol conflict.
* [Fix] fix lint errors.
* [Fix] update ops.md with info of ball_query support by MLU backend.
* [Feature] Fix typo.
* [Fix] Remove print.
* [Fix] get mlu-ops from MMCV_MLU_OPS_PATH env.
* [Fix] update MMCV_MLU_OPS_PATH check logic.
* [Fix] update error info when failed to download mlu-ops.
* [Fix] check mlu-ops version matching info in mmcv.
* [Fix] revise wrong filename.
* [Fix] remove f.close and re.
2023-01-18 14:41:04 +08:00
zcc-zjut
84f60c178c
[Feature] Add support for Ascend devices with gather_points ( #2555 )
2023-01-18 10:33:56 +08:00
mengpenghui
71ee2a61f2
[Enhance] Add AMP support for MLU_DCNv2 ( #2548 )
2023-01-13 17:51:02 +08:00
mengpenghui
c310d28c8f
[Feature] Add MLU support for DCN ( #2540 )
2023-01-13 17:46:19 +08:00
sherie
2810718a99
[Feature] Add Ascend support for RoIPool op ( #2483 )
...
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
2023-01-12 11:52:28 +08:00
ZShaopeng
48ea88ab9f
[Feature] Support Voxelization with cambricon MLU device ( #2500 )
...
* [Feature] Support hard_voxelize with cambricon MLU backend
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
* [Feature](bangc-ops): add voxelization op
2023-01-10 19:43:45 +08:00
DaGaiBa
73856344f5
[Feature] Support PSAMask op for Ascend device ( #2487 )
2023-01-06 20:15:50 +08:00
Zhongyu Li
935ba78b39
[Fix] Fix skip_layer for RF-Next ( #2489 )
...
* judge skip_layer by fullname
* lint
* skip_layer first
* update unit test
2022-12-28 15:27:34 +08:00
jayggh
46eb9ec5d0
[Feature] Add NPU adapter for Deformable RoIPool ( #2481 )
...
* add npu adapter for deformroipool
* cleancode
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
2022-12-14 15:09:46 +08:00
jayggh
4c51afce2a
[Feature] Add FusedBiasLeakyRelu npu adapter ( #2474 )
...
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* clean code
* fix autocast bugs on npu (#2273 )
fix autocast bugs on npu (#2273 )
* code format
* code format
* code format
* bug fix
* pytorch_npu_helper.hpp clean code
* Npu dev (#2306 )
* fix autocast bugs on npu
* using scatter_kwargs in mmcv.device.scatter_gather
* raise ImportError when compile with npu
* add npu test case (#2307 )
* add npu test case
* Update focal_loss.py
* add comment
* clean lint
* update dtype assert
* update DDP forward and comment
* fix bug
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* sigmoidfocalloss npu adapter bug fix
* BugFix: modify softmaxFocalLoss adapter
* BugFix: remove equal sign in the code
* add npu install information in README
* add modulatedDeformConv npu adapter
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* add modulatedDeformConv npu adapter
* merge master branch 20221103
* Add masked_ Conv2d operator in NPU
* add nms_npu
* fix bug
* fix code check
* fix code check
* fix code check
* Masked_conv2d NPU
* Masked_conv2d NPU
* Masked_conv2d NPU
* remove npu-install-info in README.md
* annotate the clang-format in pre-commit-config-zh-ch.yaml
* Clean code: fix the clean code problem in masked_conv2d and modulated_deform_conv
* Create fused_bias_leakyrelu_npu.cpp
Add NPU adapter for fused_bias_leaky_relu operator
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
* Update ops.md
* Update ops.md
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
* Update test_fused_bias_leakyrelu.py
* Update fused_bias_leakyrelu.py
* Update test_fused_bias_leakyrelu.py
* Update fused_bias_leakyrelu.py
* Update test_fused_bias_leakyrelu.py
* Update ops.md
* amend for CI
* bugfix
* amend ops.md
* Update test_fused_bias_leakyrelu.py
* clean code
* bugfix
* clean code
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
Co-authored-by: wangjiangben <wangjiangben_hw@126.com>
Co-authored-by: ckirchhoff <515629648@qq.com>
Co-authored-by: wangjiangben-hw <111729245+wangjiangben-hw@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zcc-zjut <zcczxy2019@163.com>
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
Co-authored-by: momo609 <963372609@qq.com>
2022-12-13 11:13:46 +08:00
Zaida Zhou
4e67e481a0
[Fix] Fix ut caused by the new version of packaging ( #2482 )
2022-12-11 17:59:11 +08:00
mengpenghui
49d891754b
[Feature] Support modulated_deform_conv with cambricon MLU backend ( #2411 )
...
* [Feature] Support modulated_deform_conv with cambricon MLU backend
* fix error of torch_mlu
* modify with commit suggestion
* Update modulated_deform_conv.py
* Update mmcv/ops/modulated_deform_conv.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-04 17:46:44 +08:00
SUNYC
c1712ee290
[Enhance] Add conf_path in PetrelBackend ( #2459 )
...
* feat: add conf_path in PetrelBackend
* fix: add conf_path in MockPetrelClient
* fix: adjust parameters order
2022-11-28 11:48:13 +08:00
Shanghua Gao
6e3827d02b
[Feature] Support receptive field search of CNN models ( #2056 )
...
* support rfsearch
* add labs for rfsearch
* format
* format
* add docstring and type hints
* clean code
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* rm unused func
* update code
* update code
* update code
* update details
* fix details
* support asymmetric kernel
* support asymmetric kernel
* Apply suggestions from code review
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Apply suggestions from code review
* Apply suggestions from code review
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Apply suggestions from code review
* Apply suggestions from code review
* Apply suggestions from code review
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Apply suggestions from code review
* add unit tests for rfsearch
* set device for Conv2dRFSearchOp
* Apply suggestions from code review
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* remove unused function search_estimate_only
* move unit tests
* Update tests/test_cnn/test_rfsearch/test_operator.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Apply suggestions from code review
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/cnn/rfsearch/operator.py
Co-authored-by: Yue Zhou <592267829@qq.com>
* change logger
* Update mmcv/cnn/rfsearch/operator.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: lzyhha <819814373@qq.com>
Co-authored-by: Zhongyu Li <44114862+lzyhha@users.noreply.github.com>
Co-authored-by: Yue Zhou <592267829@qq.com>
2022-11-22 19:15:55 +08:00
ZShaopeng
12442667ff
[Feature] Support MultiScaleDeformableAttn with cambricon MLU backend ( #2396 )
...
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support ms with cambricon MLU backend
* [Feature] Support msdeformattn_1104 with cambricon MLU backend
* [Feature] Support ms with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* Revert "[Feature] Support MsDeformAttn with cambricon MLU backend"
This reverts commit 27963cccc86d240852a40a2c1510147a3e9f269f.
* [Feature] Support MsdeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsdeformAttn with cambricon MLU backend
Co-authored-by: zhangshaopeng <wicky-zheng@outlook.com>
Co-authored-by: wicky-zheng <root@notebook-mmcv-290m8-torch1-6-y0yv66-notebook-0.notebook-mmcv-290m8-torch1-6-y0yv66.ns-ad8b689a0ecd41fcb4469c803dcd539d.svc.cluster.local>
2022-11-16 14:08:04 +08:00
ckirchhoff
de6b0021af
[Feature] Support masked_conv in Ascend device ( #2387 )
...
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* clean code
* fix autocast bugs on npu (#2273 )
fix autocast bugs on npu (#2273 )
* code format
* code format
* code format
* bug fix
* pytorch_npu_helper.hpp clean code
* Npu dev (#2306 )
* fix autocast bugs on npu
* using scatter_kwargs in mmcv.device.scatter_gather
* raise ImportError when compile with npu
* add npu test case (#2307 )
* add npu test case
* Update focal_loss.py
* add comment
* clean lint
* update dtype assert
* update DDP forward and comment
* fix bug
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* sigmoidfocalloss npu adapter bug fix
* BugFix: modify softmaxFocalLoss adapter
* BugFix: remove equal sign in the code
* add npu install information in README
* add modulatedDeformConv npu adapter
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* add modulatedDeformConv npu adapter
* merge master branch 20221103
* Add masked_ Conv2d operator in NPU
* add nms_npu
* fix bug
* fix code check
* fix code check
* fix code check
* Masked_conv2d NPU
* Masked_conv2d NPU
* Masked_conv2d NPU
* remove npu-install-info in README.md
* annotate the clang-format in pre-commit-config-zh-ch.yaml
* Clean code: fix the clean code problem in masked_conv2d and modulated_deform_conv
Co-authored-by: wangjiangben <wangjiangben_hw@126.com>
Co-authored-by: wangjiangben-hw <111729245+wangjiangben-hw@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zcc-zjut <zcczxy2019@163.com>
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
Co-authored-by: momo609 <963372609@qq.com>
2022-11-15 10:23:17 +08:00
q.yao
6254ebef8d
[Fix] Create Tensor with new_* method to support amp ( #2389 )
2022-11-08 15:54:34 +08:00
Zhang
b622fb2e29
[Feature] Support RoIAwarePool3d with cambricon MLU backend ( #2359 )
...
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
2022-11-03 11:13:34 +08:00
VVsssssk
93fe4829f7
[Features] Add stack ball query and stack group points ops ( #2292 )
...
* add stack sa model ops
* fix lint
* fix lint
* fix comments
* fix bug
* fix lint
* fix comments
* fix lint
* fix lint
* fix
2022-10-28 16:53:33 +08:00
Yue Zhou
a4f304a5f5
[Feature] Add box_iou_quadri & nms_quadri ( #2277 )
...
* update
* update
* fix link
* fix bug
* update nms_quadri
* fix lint
* Update test_nms_quadri.py
* Update box_iou_quadri.py
* fix bug
* Update test_nms_quadri.py
* Update box_iou_rotated_utils.hpp
* Update box_iou_quadri.py
* Update mmcv/ops/nms.py
2022-10-28 16:53:33 +08:00
takuoko
9709ff3f8c
[Enhancement] Add a new argument define_metric in wandb hook ( #2237 )
...
* wandb define_metric
* add test and some fix based on mmengine PR
* fix test
* add summary warnings
* Update mmcv/runner/hooks/logger/wandb.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/runner/hooks/logger/wandb.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-10-24 23:13:07 +08:00
cathyzhang222
7fd7058a9c
[Feature] Support Iou3d with cambricon MLU backend ( #2339 )
...
* [Feature] Support Iou3d with cambricon MLU backend
* [Feature] add double line
* [Feature] add line
2022-10-24 15:58:58 +08:00
q.yao
6cf9ac0507
[Fix] Fix three nn op can not accept half tensor ( #2348 )
...
* Fix three nn half inpt
* update test
2022-10-24 11:15:21 +08:00
Ivan Shcheklein
e417035f5d
[Enhancement] Add ability to pass logger instance to frameworks ( #2317 )
...
* Add ability to pass logger instance to frameworks
* refine docstring
* Update mmcv/runner/hooks/logger/dvclive.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-10-12 12:09:53 +08:00
CokeDong
dd5b415d2a
[Feature] Support PrRoI op for Parrots ( #2280 )
...
* Support parrots extension for op PrRoI
* Fix lint
* Fix cpp lint
* Fix testcase failure by false requires_grad in self-defined autograd Funtion
* Fix issues
* Fix flake8
* Fix isort
* Adaption for typechecking for PrRoIPoolFunction
* Fix lint
* Support only float32
* bugfix
2022-10-11 15:21:00 +08:00
q.yao
c001e2fcba
[Refactor] Refactor the nms3d op to support MLU ( #2296 )
...
* update iou3d
* fix parrot
* update to device
* replace count_nonzero with fill
* update build.yml
2022-10-11 10:24:46 +08:00
Zaida Zhou
b6a7fd98e4
Upgrade pre commit hooks ( #2321 )
...
* Upgrade the versions of pre-commit hooks
* update the versions of zh-cn.yaml
2022-10-08 11:48:44 +08:00
wangjiangben-hw
a5db5f6682
[Feature] Support training on NPU device ( #2262 )
...
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* clean code
* fix autocast bugs on npu (#2273 )
fix autocast bugs on npu (#2273 )
* code format
* code format
* code format
* bug fix
* pytorch_npu_helper.hpp clean code
* Npu dev (#2306 )
* fix autocast bugs on npu
* using scatter_kwargs in mmcv.device.scatter_gather
* raise ImportError when compile with npu
* add npu test case (#2307 )
* add npu test case
* Update focal_loss.py
* add comment
* clean lint
* update dtype assert
* update DDP forward and comment
* fix bug
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: ckirchhoff <515629648@qq.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-09-30 21:05:37 +08:00
takuoko
afff388692
[Enhancement] Support SiLU with torch < 1.7.0 ( #2278 )
...
* support silu torch<1.7.0
* fix test
* fix test
* fix inplace
* fix inplace
* Update tests/test_cnn/test_silu.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-09-29 15:19:41 +08:00
takuoko
55c51e1734
[Feature] Register a new activatation layer SiLU to ACTIVATION_LAYERS ( #2256 )
2022-09-13 22:38:23 +08:00
ZShaopeng
83bedf5a2b
[Feature] Support RoipointPool3d with cambricon MLU backend ( #2247 )
...
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
* [Feature] Support RoipointPool3d with cambricon MLU backend
2022-09-08 17:19:12 +08:00
bdf
e843d73fc7
[Feature] Support DeformRoiPool with cambricon MLU backend ( #2137 )
...
* [Feature] Support DeformRoiPool with cambricon MLU backend
* [Fix] Remove use of std library
* [Fix] Correct the error information
* [Refactor] Refactor test deform_roi_pool code
* [Fix] Fix judgment error
* [Fix] Modify the large tensor check
Co-authored-by: budefei <budefei@cambricon.com>
2022-08-28 22:37:28 +08:00
liuduanhui
a08517790d
[Feature] Support ThreeNN with cambricon MLU backend ( #2215 )
2022-08-23 15:18:47 +08:00
bdf
e663670a74
[Feature] Support MaskedConv2d with cambricon MLU backend ( #2202 )
...
* [Feature] Support MaskedConv2d with cambricon MLU backend
* [Refactor] Refactor test masked_conv2d code
Co-authored-by: budefei <budefei@cambricon.com>
2022-08-21 23:21:54 +08:00
bdf
832b845257
[Feature] Add carafe op for MLU ( #2212 )
...
* [Feature] Support CARAFE with Cambricon MLU backend
* [Docs] Add comments for common funtions
* [Test] Add allclose test for carafe
* Remove print
Co-authored-by: zcyKTH <zcy19950525@gmail.com>
Co-authored-by: budefei <budefei@cambricon.com>
2022-08-19 10:36:26 +08:00
Jingwei Zhang
b9bdfeb6f6
[Enhance] Support different border modes in imrotate ( #2141 )
...
* support different border_mode in imrotate
* fix docstring
* small fix
* reorder the border_mode arg in imrotate
* fix lint
* polish docstring
* polish docstring
2022-08-17 10:53:05 +08:00
Yosuke Shinya
304f184c46
[Fix] Fix gradient cumulative optimizer when resuming ( #2093 )
...
* Fix gradient cumulative optimizer when resuming
* add unit test
2022-08-15 10:50:43 +08:00
Yixiao Fang
b2ac245602
[Refactor] Add pillow backend for ColorJitter related functions ( #2127 )
...
* add 'backend' for ColorJitter related functions
* add unittest
* fix unittest
* add bgr transpose and revise ut
* update unittest
* revise docstring
2022-08-03 16:21:41 +08:00
xcnick
4558bfbd5b
[Fix] Fix onnx unit tests ( #2155 )
2022-07-29 19:23:29 +08:00
Peter Ye
78f01001d5
[Feature] Support setting start epoch and interval epoch for PAVI Logger Hook ( #2103 )
...
* [Feature] Support setting start epoch and interval epoch for PAVI Logger Hook
* [Feature] Update the coding style as the maintainer wish
* fix: default integer division or modulo by zero
* fix: runner.epoch is less than start and use self.get_epoch instead of runner.epoch
* feat: support for iter-based runner and fix the step bug
* feat: iter based hook
* feat: fix bug and coding style
* fix: coding style
* fix: coding style
* fix: graph may add in evaluation
2022-07-29 11:40:20 +08:00
Jingwei Zhang
35affc35bf
[Feature] Support PrRoIPool operation ( #2113 )
...
* support prroipooling
* fix build bug
* fix bug of precision
* fix lint
* add ut
* add license and citation
* polish op and add type hint
* refactor ut
* add docstring for prroipool
* polish code
* format cu and cuh code
* fix format
* fix format
* fix typo
Co-authored-by: zhouzaida <zhouzaida@163.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-07-23 23:08:32 +08:00