Commit Graph

312 Commits (4150d0ac2f4e46572f0b16d51b2658700b0c2bdc)
 

Author SHA1 Message Date
Kai Chen c728193f8d bug fix for the latest torchvision 2019-05-09 00:16:04 -07:00
Kai Chen e2e109e4a5
Merge pull request #66 from wangg12/master
add bgr2hls and hls2bgr
2019-05-06 21:28:15 -07:00
Gu Wang d4b91b8376 test bgr2hls 2019-05-07 12:04:01 +08:00
Gu Wang a092de341c add bgr2hls and hls2bgr 2019-04-30 22:52:14 +08:00
Kai Chen 0d77f61450
Merge pull request #64 from ycxioooong/master
video fps should be float
2019-04-23 21:06:24 -07:00
xiongyu d8bdc8679c video fps should be float 2019-04-24 11:58:50 +08:00
Kai Chen 5d88696a8a
Merge pull request #63 from hellock/master
Add jhu gn model urls and update to v0.2.7
2019-04-23 18:07:54 -07:00
Kai Chen bb089030a9 remove wrong license argument 2019-04-23 12:52:02 -07:00
Kai Chen cc207c4523 add jhu gn model urls and update to v0.2.7 2019-04-23 00:15:27 -07:00
Kai Chen 4abb916d60
Merge pull request #60 from wangg12/master
fix possible encoding error when install
2019-04-21 19:49:46 -07:00
Gu Wang 11ad40b560 fix possible encoding error when install 2019-04-22 10:43:33 +08:00
Kai Chen d99c6f8ddf
Merge pull request #62 from hellock/collections-abc
Use collections.abc instead of collections
2019-04-21 19:33:54 -07:00
Kai Chen 528bf72ce8
Merge pull request #61 from hellock/weight-init
Add caffe2_xavier_init()
2019-04-21 19:16:24 -07:00
Kai Chen b7c924cb96 use collections.abc instead of collections 2019-04-21 19:13:28 -07:00
Kai Chen 5e3d09bc5d add caffe2_xavier_init() because of different implementations of caffe2 and pytorch 2019-04-21 18:51:38 -07:00
Yue Zhao 944da49fcb add iminvert (#58)
* add iminvert

* fix for flake8

* minor improvement

* add test for iminvert

* fix for flake8

* fix for flake8

* revise __init__

* revise __init__
2019-04-20 00:27:59 -07:00
Kai Chen 38fa314226
Add GN+WS pretrained model urls (#59)
* add GN+WS pretrained model urls

* fix linting errors
2019-04-19 21:27:34 -07:00
Cao Yuhang 8266d877fe add json log support (#55)
* add json log support

* add some comment, minor fix

* rename variable vv to item

* mv round float to a function
2019-04-19 00:18:33 -07:00
Kai Chen 506455af0b
Update version to 0.2.6 (#54)
* fix the time estimation when resuming from a checkpoint

* fix the time estimation when resuming from a checkpoint

* support str type for loggers

* fix yaml dump tests

* update version to v0.2.6
2019-04-14 22:55:58 -07:00
Cao Yuhang b85136772d fix bug that stuck at evaluation (#53)
* fix bug that stuck at evaluation

* remove mode variable
2019-04-14 22:28:05 -07:00
Kai Chen bef7c13a95
Merge pull request #47 from yhcao6/memory_statistic
reduce max memory in dist training
2019-04-10 22:38:45 -07:00
Cao Yuhang 5726795c7a reduce max memory in dist training 2019-04-08 14:07:43 +08:00
Kai Chen 4a1ddfc8a1
Merge pull request #44 from yhcao6/master
fix memory order in log text
2019-04-06 19:32:53 -07:00
Cao Yuhang f3d5d97919 fix memory order in log text 2019-04-07 10:20:52 +08:00
Cao Yuhang 1263c91e5c add memory statistic (#41)
* add memory statistic

* fix import order and cuda check
2019-04-05 23:11:35 -07:00
Kai Chen a0ce998ecd update to v0.2.5 2019-03-25 22:54:42 -07:00
Kai Chen d426585999
Support str type for loggers (#39)
* fix the time estimation when resuming from a checkpoint

* fix the time estimation when resuming from a checkpoint

* support str type for loggers

* fix yaml dump tests
2019-03-25 21:59:02 -07:00
Aitical b0f40396ad Update the documentation (#36)
* update the documentation

* update the documentation

* Update io.md
2019-03-09 16:16:08 -08:00
Kai Chen 76bcbe2d5e
Merge pull request #38 from wangg12/master
allow .yml extension for config file
2019-03-09 15:36:46 -08:00
wangg12 bfdd0d5985 allow .yml extension for config file 2019-03-08 10:08:58 +08:00
Kai Chen 8b4417c19a
Fix the time estimation when resuming from a checkpoint (#37)
* fix the time estimation when resuming from a checkpoint

* fix the time estimation when resuming from a checkpoint
2019-03-02 16:16:39 -08:00
Kai Chen 1d6e91b1b0
Merge pull request #35 from wangg12/master
add cosine lr schedule
2019-03-01 20:50:45 -08:00
wangg12 ccb7ca30f8 add cosine lr schedule 2019-02-16 17:42:27 +08:00
Kai Chen c8c3493868
Merge pull request #33 from OneDirection9/master
bug fixes
2019-02-12 18:08:00 +08:00
HanZhipeng 2882ef12ce
bug fixes 2019-02-11 16:47:16 +08:00
Kai Chen 8a3e232cb4
Merge pull request #32 from youkaichao/typo
correct typo
2019-02-10 22:14:32 +08:00
youkaichao e4e6960a8c correct typo 2019-02-10 19:46:48 +08:00
pangjm 841d4d4092 add resnext50_32x4d pretrain model 2019-01-24 20:05:44 +08:00
Kai Chen 0f7d5718db
Merge pull request #31 from tjsongzw/master
fix typo
2019-01-23 19:45:15 +08:00
tjsongzw 68b6887ea0 fix typo 2019-01-23 18:02:47 +08:00
Kai Chen 0ad619c17a update to v0.2.4 2019-01-17 14:17:20 +08:00
Kai Chen 2415d46b9e add gn model urls 2019-01-14 17:39:08 +08:00
Kai Chen 8f96c2a4bb
Merge pull request #29 from wangg12/master
allow passing custom logger to runner
2019-01-07 22:46:16 +08:00
wangg12 7b79237d26 update doc 2019-01-07 22:09:35 +08:00
wangg12 0f88eddb62 allow passing custom logger to runner 2019-01-06 15:17:35 +08:00
Kai Chen ddb5222817
Merge pull request #28 from open-mmlab/pytorch-1.0
Add compatibility with PyTorch 0.4.1 and 1.0
2018-12-25 15:48:57 +08:00
Kai Chen 477010fb87 add a blank line 2018-12-25 15:39:47 +08:00
Kai Chen afe97d53d6
Merge branch 'master' into pytorch-1.0 2018-12-25 15:19:33 +08:00
Kai Chen b64d43873d make compatible with pytorch 0.4.1 and 1.0 2018-12-25 15:13:44 +08:00
Kai Chen fd009613d3 update to 0.2.2 2018-12-19 19:23:58 +08:00