NanoCode012
901243c780
Fix relative path for VOC download ( #880 )
...
Cleaned up code
2020-08-31 16:06:10 -07:00
NanoCode012
1e15aad6f9
Add InfiniteDataLoader class ( #876 )
...
* Add InfiniteDataLoader
Only initializes at first epoch. Saves time.
* Moved class to a better location
2020-08-31 11:01:25 -07:00
NanoCode012 and Glenn Jocher
fb4fc8cd02
Fix ema attribute error in DDP mode ( #775 )
...
* Fix ema error in DDP mode
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2020-08-18 09:52:21 -07:00
NanoCode012 and Glenn Jocher
0892c44bc4
Fix Logging ( #719 )
...
* Add logging setup
* Fix fusing layers message
* Fix logging does not have end
* Add logging
* Change logging to use logger
* Update yolo.py
I tried this in a cloned branch, and everything seems to work fine
* Update yolo.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com >
2020-08-12 14:18:19 -07:00
NanoCode012 and Glenn 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
NanoCode012 and Glenn 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
NanoCode012 and Glenn 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
NanoCode012 and Glenn 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
NanoCode012