Commit Graph
7 Commits
Author SHA1 Message Date
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
Liu ChangyuandGlenn Jocher 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
Lornatang 899f1d4bde Fix DDP bug in single process multiple device use cases 2020-06-20 13:00:03 +08:00
Lornatang 2368603484 fix refrence bug
In torch==1.5, the import of the API has changed. Although it does not interrupt the operation of the program, it seems to me to be an implicit error and may throw an exception in later versions.
2020-06-17 09:56:26 +08: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