mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
495 Commits
13 Branches
108 Tags
42 MiB
Commit Graph

495 Commits (ed2887bb72b7c8cee11aba94663831b8f6bd8827)
 

Author SHA1 Message Date
Wenwei Zhang 7a6285b190
Fix mmcv-dataparallel (#497) 
* 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
 2020-08-19 14:22:40 +08:00
Jintao Lin 4ec73abbcc
Logging mode by using runner.mode (#495) 
* log mode by using runner.mode

* add IterTimerHook to fix unittest

* correct the logic

* display fix when using EvalHook in train mode

* simplify logic
 2020-08-17 18:50:19 +08:00
Cao Yuhang 7b18b97791
fix saconv (#489) 
* fix saconv

* add parrots condition

* add unittest

* fix torch version
 2020-08-16 01:45:01 +08:00
Wang Xinjiang eacaf475f7
fix some pavi logger hooks (#481) 
* fix some pavi logger hooks

* fix unittest

* fix small bugs

* small change

* fix unittest

* Add EpochBasedRunner conditions

* Add session text

* fix small bug

* fetch runner mode from log buffer

* Add max_iter to pavi session text

* change yaml.dump to yamp.dump(yaml.load(mmcv.dump))

* Directly use by_epoch

* fix unittest

* add comments

* Use runner.epoch + 1 in pavi log

* fix runner.epoch issue for runner.mode=='val'

* fix runner.epoch issue for runner.mode=='val'

* Use abspath instead of realpath

* Add meta dump unittest

* small change

* Add comments
 2020-08-16 01:20:08 +08:00
Jintao Lin c8e85b28e4
Add `tin_shift` function (#492) 
* add tin shift

* add unittest

* add docstring

* add docstring

* parrots for tin_shift

* fix lint

* fix lint

Co-authored-by: jiaomenglei <jiaomenglei@sensetime.com>
 2020-08-15 22:29:07 +08:00
Jerry Jiarui XU 15537c5a39
Add LOCAL_RANK env var for slurm (#496) 2020-08-14 20:35:05 +08:00
Wang Xinjiang d19fad02ba
Add json set_default for unserializable values (#485) 
* Add json set_default for unserializable values

* kwargs.setdefault
 2020-08-13 22:05:43 +08:00
Cao Yuhang dc778481cb
add op trouble shooting (#479) 
* add op trouble shooting

* update trouble_shooting.md

* clean ops.md

* add trouble shooting to index.rst

* reorder

* add troubleshooting in readme
 2020-08-13 22:04:58 +08:00
Wenwei Zhang 5ade35f4cf
use elif rather than if (#491) 2020-08-13 20:38:27 +08:00
Jerry Jiarui XU 4cc48073bc
Change pretrain url from aws to aliyun (#490) 2020-08-13 19:32:28 +08:00
Zijian He e7e0c89f5c
fix the bug (#472) (#484) 
* fix the bug (#472)

* fix the bug (#472)

* fix the bug (#472)

Co-authored-by: hezijian <hezijian@dm-ai.cn>
 2020-08-13 19:30:28 +08:00
Yuanhao Zhu 51c65c97ec
fix syncbn parameter order mismatch and parrots bug (#488) 2020-08-13 19:27:05 +08:00
su 17e4732c49
Change the epoch runner to use the data_loader from attributes rather than args. (#483) 2020-08-12 22:09:07 +08:00
Kai Chen e92f826abc
close the config file after opening (#480) 2020-08-11 21:28:44 +08:00
Kai Chen cac22f8cf5
Add pytorch 1.6 to CI (#476) 
* add pytorch 1.6 to CI

* fix typo

* fix ci error

Co-authored-by: Cao Yuhang <yhcao6@gmail.com>
 2020-08-09 21:56:15 +08:00
Jerry Jiarui XU 56e71a718b
Add Depthwise Seperable ConvModule (#477) 2020-08-09 17:29:02 +08:00
Kai Chen 530ae20046
add pre-build packages for pytorch 1.6 (#475) 2020-08-09 15:44:44 +08:00
Jerry Jiarui XU 9b36c1ab3e
Fixed iter_base_runner logger (#474) 2020-08-09 00:04:48 +08:00
Jintao Lin edbbc0f215
fix bug for self.warmup_iters when warmup_by_epoch (#470) 2020-08-06 22:07:31 +08:00
ChaseMonsterAway 6b43ae3700
Update cnn.md (#466) 2020-08-06 16:51:58 +08:00
Danil 0a1cea73c1
Add dot to yml extension (#469) 2020-08-06 16:51:32 +08:00
Kai Chen 045629c280
Fix torch installation in github action (#462) 
* bump version to 1.0.5

* fix pytorch name

* fix version

* disable the cache
 2020-08-03 23:10:39 +08:00
Wenwei Zhang d36850f517
Add git hash (#461) 
* Add git hash

* Add fallback

* Add func in mmcv.utils
 2020-08-03 21:00:33 +08:00
Kai Chen 60ce48d164
bump version to 1.0.5 (#459) 2020-08-03 13:17:03 +08:00
Jintao Lin 71b59a8da0
assign `self.warmup_iters` in `before_run` function (#458) 2020-08-03 11:55:51 +08:00
Jintao Lin 9de04d2284
Replace `re` with `regex` (#457) 
* replace `re` with `regex`

* add regex in requirements

* use regex on windows and re on linux

* update requirements.txt
 2020-08-03 11:45:50 +08:00
Jerry Jiarui XU 66864d4abc
Update pretrain link (#455) 
* update pretrain link

* remove some
 2020-08-02 22:00:05 +08:00
shilong 1830347f8b
Ema (#421) 
* add ema hook

* add ema hook resume

* add ema hook test

* fix typo

* fix according to comment

* delete logger

* fix according to comment

* fix unitest

* fix typo

* fix according to comment

* change to resume_from

* typo

* fix isort
 2020-07-30 22:06:19 +08:00
Cao Yuhang 4676031c8b
Move onnx op import inside symbolic (#453) 
* add pat

* move onnx op import to symbolic
 2020-07-28 23:14:36 +08:00
Jintao Lin 829d1e381d
add function and class related with fp16 to runner/__init__.py (#448) 2020-07-27 10:43:20 +08:00
Wang Xinjiang d4da3daa7e
Syncbuf (#447) 
* More robust sync buffer hook

* More robust sync buffer hook

* Reformat
 2020-07-25 12:51:46 +08:00
Kai Chen cec5aacefb
bump version to 1.0.4 (#446) 2020-07-24 22:03:15 +08:00
Jiamin b20d6ab18b
Update jenkinsfile (#444) 
* chore: use https and rm work_dir before cp

* chore: explictly install Cython

* chore: rm dir:tmp before cp
 2020-07-24 19:01:35 +08:00
Wang Xinjiang 66604e83de
Add syncbuffer hook (#443) 
* reformat

* reformat

* Add register hook from cfg

* docstring

* change according to comments
 2020-07-24 14:15:44 +08:00
robin Han 903091effc
add __init__.py so that we can find onnx_util while packaging (#442) 2020-07-24 12:00:05 +08:00
Jiamin fad12cd366
Add jenkinsfile (#415) 
* chore: add jenkins file

* chore: remove cuda_arch

* chore: replace Float.parsefloat with string check

* chore: add pip mirror

* chore: fix add MMCV_WITH_OPS env var

* chore: fix undefined symbol _Z16THPVariable_WrapN2at6TensorE

* chore: add comment

* chore: fix multi docker running under same dir error

* chore: rm dir after build

* chore: fix rm

* chore: also rm tmp dir

* chore: use sensetime docker

* feat: use torch-pre-installed docker

* fix: fix index error
 2020-07-23 17:06:34 +08:00
Kai Chen 487f381f25
fix docs (#440) 2020-07-21 14:48:16 +08:00
Jiamin 55fadb4c4e
Add runner.meta to checkpoint in save_checkpoint() (#438) 
* fix: error when runner.meta is None

* tests: add unittest for epoch-based save_checkpoint
 2020-07-20 11:40:04 +08:00
Yang Li 215a324414
Support update specific steps (#437) 2020-07-20 11:36:00 +08:00
sandyhsia 82e5c8e552
fix: close temp file after creating it to avoid permssion problem on windows (#436) 2020-07-20 00:16:39 +08:00
Kai Chen 3f936167de
bump version to 1.0.3 (#434) 2020-07-17 23:48:59 +08:00
Yawei Li 7730a79fcd
fix typo of annealing (#433) 2020-07-17 23:48:22 +08:00
Jintao Lin 6ece0e5d19
Add pairwise function for 'gaussian' and 'concatenation' mode in NonLocal. (#424) 
* add pairwise function for 'gaussian' and 'concatenation' mode

* rename test function

* decrease the complexity of nonlocal unittest

* fix typo and make unittest more complete

* add unittest when zero_init is False

* minor fix

* pack theta and phi

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
 2020-07-17 23:44:50 +08:00
Jerry Jiarui XU 49fdf3cfa0
add FP16 optimizer (#432) 2020-07-17 21:25:47 +08:00
Jerry Jiarui XU 9e3ff5f94c
Support pillow resize (#426) 
* Support pillow resize

* add more types
 2020-07-17 19:25:47 +08:00
robin Han 34ee69e0f6
support converting pytorch to onnx (#429) 
* support convert nms to onnx op

* add comment for NMSop

* add corresponding onnx support for mmdet/mmseg/mmediting

* remove onnx part from mmcv/__init__
 2020-07-17 19:24:46 +08:00
Jerry Jiarui XU 07e4215286
Fixed group of syncbn (#427) 2020-07-17 11:16:47 +08:00
Jerry Jiarui XU 4a044c6466
Support FP16 (#428) 
* Support FP16

* update ci
 2020-07-17 11:14:07 +08:00
Jerry Jiarui XU 7cfc839ea5
Add MMSyncBN in registry (#420) 
* Add MMSyncBN in registery

* skip mmsyncbn test
 2020-07-15 19:23:09 +08:00
Jiamin 50f69e7057
fix: _get_max_memory when model has no attr output_device (#416) 2020-07-15 17:37:14 +08:00