539 Commits

Author SHA1 Message Date
Glenn Jocher
93684531c6
train.py --logdir argparser addition (#660)
* train.py --logdir argparser addition

* train.py --logdir argparser addition
2020-08-06 22:26:38 -07:00
NanoCode012
886b9841c8
Add Multi-Node support for DDP Training (#504)
* Add support for multi-node DDP

* Remove local_rank confusion

* Fix spacing
2020-08-06 11:15:24 -07:00
Glenn Jocher
7eaf225d55 zero-target training bug fix (#609) 2020-08-05 13:35:31 -07:00
hopesala
d0d3dd1040
Update detect.py (#629)
* Update detect.py

Update print info for relative path and absolute
otherwise the info below is misleading

Results saved to /workspace//output_path

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-05 12:01:24 -07:00
Glenn Jocher
b8f3b1b373
Update ci-testing.yml (#637)
* Update ci-testing.yml

* Update ci-testing.yml
2020-08-05 11:42:51 -07:00
lorenzomammana
728efa6576
Fix missing imports (#627)
* Fix missing imports

* Update detect.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-04 12:22:15 -07:00
Glenn Jocher
941cb2680b Merge remote-tracking branch 'origin/master' 2020-08-03 19:55:57 -07:00
Glenn Jocher
52b626392f update README.md 2020-08-03 19:55:12 -07:00
Wang Xinyu
7ecf09d6da
Update README.md, deploy model with tensorrt (#623)
* Update README.md, deploy model with tensorrt

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-03 19:41:50 -07:00
Glenn Jocher
4ee8adb04b Merge remote-tracking branch 'origin/master' 2020-08-03 17:18:37 -07:00
Glenn Jocher
f346da9f2b update activations.py 2020-08-03 17:18:31 -07:00
Glenn Jocher
f807e7b867
Update README.md 2020-08-03 11:11:12 -07:00
Glenn Jocher
937b9db977
Update README.md 2020-08-03 11:09:46 -07:00
Glenn Jocher
eb99dff9ef import random bug fix (#614) 2020-08-03 10:52:59 -07:00
Jirka Borovec
d5b6416c87
Explicit Imports (#498)
* expand imports

* optimize

* miss

* fix
2020-08-02 15:47:36 -07:00
Glenn Jocher
ec7a926163 AutoAnchor update to display anchors/target 2020-08-02 14:38:45 -07:00
Glenn Jocher
d989bc9260 remove NBSP 2020-08-02 14:23:05 -07:00
Glenn Jocher
023e37807c hyperparameter evolution update (#566) 2020-08-02 13:18:03 -07:00
Glenn Jocher
64e6d199ba hyperparameter evolution update (#566) 2020-08-02 13:17:26 -07:00
Glenn Jocher
fdbcc8fd7f hyperparameter evolution update (#566) 2020-08-02 12:30:34 -07:00
Glenn Jocher
f1096b2cf7 hyperparameter evolution update (#566) 2020-08-02 10:32:04 -07:00
Glenn Jocher
305c6a028a compute_loss() leaf variable update 2020-08-02 10:30:43 -07:00
Glenn Jocher
c1a2a7a411 hyperparameter evolution bug fix (#566) 2020-08-01 23:00:10 -07:00
Glenn Jocher
8074745908 hyperparameter evolution bug fix (#566) 2020-08-01 19:24:14 -07:00
Glenn Jocher
e32abb5fb9 hyperparameter evolution bug fix (#566) 2020-08-01 19:15:48 -07:00
Glenn Jocher
8056fe2db8 hyperparameter evolution bug fix (#566) 2020-08-01 15:07:40 -07:00
Glenn Jocher
61b5733c36 update requirements.txt 2020-08-01 14:00:38 -07:00
Glenn Jocher
3e5e44499d update requirements.txt 2020-08-01 14:00:14 -07:00
Glenn Jocher
127cbeb3f5 hyperparameter expansion to flips, perspective, mixup 2020-08-01 13:47:54 -07:00
Glenn Jocher
6f08e8bcce add rebase.yml 2020-08-01 12:10:46 -07:00
Glenn Jocher
23d2bde01f leaf variable --single-cls training bug fix (#593) 2020-08-01 11:42:41 -07:00
Glenn Jocher
bcd452c482 replace random_affine() with random_perspective()
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-31 15:53:52 -07:00
Glenn Jocher
bb87276d80
update build_targets() (#589)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-31 14:34:13 -07:00
Glenn Jocher
8bf3cff875
PyTorch 1.6.0 update with native AMP (#573) 2020-07-31 11:10:03 -07:00
Glenn Jocher
60d5dc0294
PyTorch 1.6.0 update with native AMP (#573) 2020-07-31 10:54:08 -07:00
Liu Changyu
c020875b17
PyTorch 1.6.0 update with native AMP (#573)
* PyTorch have Automatic Mixed Precision (AMP) Training.

* Fixed the problem of inconsistent code length indentation

* Fixed the problem of inconsistent code length indentation

* Mixed precision training is turned on by default
2020-07-31 10:52:45 -07:00
Glenn Jocher
48e15be498
Update --bug-report.md 2020-07-31 00:23:54 -07:00
Glenn Jocher
c8892672a7 Funnel ReLU (FReLU) @staticmethod bug fix (#556)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-30 17:38:15 -07:00
Glenn Jocher
131782a2aa Funnel ReLU (FReLU) (#556)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-30 17:35:34 -07:00
AlexWang1900
4b074d9d9d
Funnel ReLU (FReLU) (#556)
* fix #541 #542

* Update train.py

* Add Frelu

* Update activations.py

PEP8 and format updates for commonality with models.common.Conv()

* Update activations.py

Update case

* Update activations.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-30 17:25:49 -07:00
Glenn Jocher
5414e53026
update ci-testing.yml with on: cron (#572)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-30 14:41:50 -07:00
Glenn Jocher
a91cdd2ead update stale.yml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-30 13:57:20 -07:00
Glenn Jocher
1df494e5fe Merge remote-tracking branch 'origin/master' 2020-07-30 11:31:54 -07:00
Glenn Jocher
2268f9c44f update *.yaml output layer comments 2020-07-30 11:31:47 -07:00
Laughing
4e2b9ecc7e
LR --resume repeat bug fix (#565) 2020-07-30 10:48:20 -07:00
Liu Changyu
adaf1d8ac1
Update models/hub *.yaml files for v2.0 (#546)
* fix #543 issue

* Update yolov3-spp.yaml

* Update yolov5-fpn.yaml

* Update yolov5-panet.yaml

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-29 14:02:07 -07:00
YuriRibeiro
7d78b17a69
Fix utils.py nonzero() warning (#551) 2020-07-29 12:14:11 -07:00
AlexWang1900
a209a32019
Fix bug #541 #542 (#545)
* fix #541 #542

* Update train.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-28 18:31:01 -07:00
Glenn Jocher
8ad6e53760
Update greetings.yml PR message (#519)
* Update greetings.yml

* Update greetings.yml
2020-07-28 15:48:01 -07:00
Glenn Jocher
aa1392a02e Merge remote-tracking branch 'origin/master' 2020-07-28 15:39:33 -07:00