Commit Graph
9 Commits
Author SHA1 Message Date
NanoCode012andGlenn Jocher 4949401a94 Fix redundant outputs via Logging in DDP training (#500)
* Change print to logging

* Clean function set_logging

* Add line spacing

* Change leftover prints to log

* Fix scanning labels output

* Fix rank naming

* Change leftover print to logging

* Reorganized DDP variables

* Fix type error

* Make quotes consistent

* Fix spelling

* Clean function call

* Add line spacing

* Update datasets.py

* Update train.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-11 11:18:45 -07:00
NanoCode012 3d8ed0a76b Fix missing model.stride in DP and DDP mode (#683) 2020-08-09 11:01:36 -07:00
NanoCode012 d7cfbc47ab Fix unrecognized local rank argument (#676) 2020-08-08 16:40:10 -07:00
NanoCode012andGlenn Jocher fe5b3f8712 Fix curl download on Windows (#669)
* Update ci-testing.yml

* Fix windows download

* Fix cookie and curl download issue from gdrive

* Revert "Update ci-testing.yml"

This reverts commit 7389d2238df4dd8bd3b6ddded0f12e551c8fd0b1.

* Update google_utils.py

PEP8

* Update google_utils.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-08-07 12:10:59 -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
NanoCode012andGlenn Jocher 7f8471eaeb --notest bug fix (#518)
* Fix missing results_file and fi when notest passed

* Update train.py

reverting previous changes and  removing functionality from 'if not opt.notest or final_epoch:  # Calculate mAP' loop.

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-25 10:24:39 -07:00
NanoCode012andGlenn Jocher 999804fea9 Add link to multi-gpu train tutorial (#476)
* Add link to multi-gpu train tutorial

* Update README.md

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-07-22 11:56:48 -07:00
NanoCode012 ccfa9373ae Update to remove "module" attribute check 2020-07-04 23:12:57 +07:00
NanoCode012 f83521ab9a Fix save error for multi-gpu 2020-07-04 16:50:33 +07:00