Commit Graph

146 Commits (cce95e744dad1d8cb487c34b1e641136063e77f3)
 

Author SHA1 Message Date
Glenn Jocher cce95e744d backbone as FP16, save default to FP32 2020-06-18 00:13:18 -07:00
Glenn Jocher d9b64c27c2 save ckpt in FP16 #119 2020-06-17 22:34:13 -07:00
Glenn Jocher 9fdb0fbacf AutoAnchor bug fix # 117 2020-06-17 19:51:15 -07:00
Glenn Jocher 8b6f5826bc update README.md 2020-06-17 12:58:01 -07:00
Glenn Jocher 2bfd242fa1 Merge remote-tracking branch 'origin/master' 2020-06-17 12:57:03 -07:00
Glenn Jocher b7ac446d62 update README.md 2020-06-17 12:56:47 -07:00
Glenn Jocher 91de3381e1
Merge pull request #112 from imprashr/patch-1
onnx should be 1.6.0
2020-06-17 08:23:35 -07:00
Prashant Rai 0071b69994
onnx should be 1.6.0
for smoothly export model to onnx, onnx should be 1.6.0
2020-06-17 14:15:27 +05:30
Glenn Jocher a557b7d924 check_img_size() update 2020-06-16 17:48:01 -07:00
Glenn Jocher 034609414e AutoAnchor fitness to screen 2020-06-16 17:08:45 -07:00
Glenn Jocher ef75f7313d AutoAnchor n, imgsz bug fix 2020-06-16 16:48:01 -07:00
Glenn Jocher 2daa412575 train with multi-gpu half test bug fix #99 2020-06-16 16:28:25 -07:00
Glenn Jocher d1e5716610 train with multi-gpu half test bug fix #99 2020-06-16 16:22:58 -07:00
Glenn Jocher 1dfc28527f AutoAnchor BPR to 4 significant figures 2020-06-16 16:19:58 -07:00
Glenn Jocher 8db51c7002 tb_writer bug fix 2020-06-16 16:05:28 -07:00
Glenn Jocher 4052603e1f AutoAnchor update - improvement check 2020-06-16 15:54:20 -07:00
Glenn Jocher afe1df385b dist.destroy_process_group() bug fix 2020-06-16 15:08:14 -07:00
Glenn Jocher 72d06147b2 Multi-GPU disabled in test.py 2020-06-16 14:42:30 -07:00
Glenn Jocher bafbc65ee3 AutoAnchor bug fix 2020-06-16 14:26:02 -07:00
Glenn Jocher 57a0ae3350 AutoAnchor implementation 2020-06-16 13:12:22 -07:00
Glenn Jocher 95c46f7245
Merge pull request #79 from Lornatang/delete-redundant-thrid-party-library
Delete redundant thrid party library
2020-06-16 11:52:36 -07:00
Glenn Jocher 5a50491fa1 check_anchors bug fix 2020-06-16 10:36:35 -07:00
Glenn Jocher 05b8ee5ca4 check_anchors() bug fix #102 2020-06-16 10:34:16 -07:00
Glenn Jocher 7c6fd4d670 --verbose bug fix #84 2020-06-16 10:22:22 -07:00
Glenn Jocher ec81c7b5f2 check_anchors() bug fix #90 2020-06-16 10:14:04 -07:00
Glenn Jocher e1e3399286 ONNX export bug fix #93 2020-06-16 09:59:42 -07:00
Glenn Jocher 8b26e89006 AutoAnchor bug fix #72 2020-06-16 00:53:34 -07:00
Glenn Jocher 8fa3724072 kmean_anchors() update 2020-06-16 00:40:21 -07:00
Glenn Jocher bdd9fee841 update fast mode 2020-06-16 00:29:54 -07:00
Glenn Jocher 1c0b6236e3 update fast mode 2020-06-16 00:11:29 -07:00
Lornatang 8b8b792802 There is no need to download extra packages, official bring it with you
I submitted it once in your yolov3 project, you seem to accept it? I'm not sure. I'll submit PR again.
2020-06-16 08:28:52 +08:00
Lornatang 0c4b4b8817 Merge remote-tracking branch 'upstream/master' 2020-06-16 08:16:10 +08:00
Lornatang 09c1b96136 Merge branch 'fix-reference-bugs' 2020-06-16 08:15:10 +08:00
Glenn Jocher db2c3acd3a updated testing settings, rebalanced towards FP16 latency 2020-06-15 16:55:57 -07:00
Glenn Jocher 915b1481fc default check_git_status() to True 2020-06-15 16:18:46 -07:00
Glenn Jocher 0b514daced FP16 test loss bug fix 2020-06-15 15:26:29 -07:00
Glenn Jocher a1748a8d6e test during training default to FP16 2020-06-15 15:20:27 -07:00
Glenn Jocher bd3e389130 update google_utils.py 2020-06-15 14:39:04 -07:00
Glenn Jocher 76ca367a01 FP16 to FP32 ckpt load 2020-06-15 13:32:45 -07:00
Glenn Jocher 6fb5ff014c FP16 to FP32 ckpt load 2020-06-15 13:21:34 -07:00
Glenn Jocher 14523bb030 FP16 to FP32 ckpt load 2020-06-15 13:18:39 -07:00
Glenn Jocher c5966abba8 glob search bug fix #77 2020-06-15 12:08:57 -07:00
Glenn Jocher ace56eb5b6
Merge pull request #70 from Lornatang/fix-reference-bugs
fix reference problem
2020-06-15 11:18:17 -07:00
Lornatang 54795a4f0a fix problem 2020-06-15 16:11:06 +08:00
Glenn Jocher 45a17ea40c Merge remote-tracking branch 'origin/master' 2020-06-15 00:44:08 -07:00
Glenn Jocher e7c24e04ac half() fix #66 2020-06-15 00:43:57 -07:00
Glenn Jocher 6570336284
Merge pull request #65 from Ashing00/master
add "__init__.py" file in <models> folder
2020-06-15 00:34:54 -07:00
ashing 62eda5e9c2 add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
Glenn Jocher 111994956e switch default inference to FP16 on GPU 2020-06-14 19:25:39 -07:00
Glenn Jocher 157aff2854 FP16 inference fix 2020-06-14 18:45:20 -07:00