Commit Graph

155 Commits

Author SHA1 Message Date
Glenn Jocher c16b587ac7
Update train.py 2019-03-25 15:06:22 +01:00
Glenn Jocher 06d264198c
Update train.py 2019-03-25 15:05:35 +01:00
Glenn Jocher c7192f64c9
Update train.py 2019-03-25 15:03:13 +01:00
Glenn Jocher cd51e1137b
Add collate_fn() to DataLoader (#163)
Multi-GPU update with custom collate function to allow variable size target vector per image without needing to pad targets.
2019-03-25 14:59:38 +01:00
Fatih Baltacı 9208a91095
Update train.py 2019-03-25 15:22:05 +03:00
perry0418 5daea5882f
Update train.py
fix problem of multiple gpu training
2019-03-25 16:13:21 +08:00
perry0418 386835d7ca
Update train.py
solve the multi-gpu training problem.
2019-03-25 14:56:38 +08:00
Glenn Jocher b31f8fb017 updates 2019-03-22 15:08:03 +02:00
Glenn Jocher 943db40f1a updates 2019-03-22 00:41:43 +02:00
Glenn Jocher 20beee0c5b updates 2019-03-21 22:51:22 +02:00
Glenn Jocher 176851f83a updates 2019-03-21 22:49:57 +02:00
Glenn Jocher 6aef4e6a78 updates 2019-03-21 22:41:12 +02:00
Glenn Jocher d047062074 updates 2019-03-21 17:26:31 +02:00
Glenn Jocher a3067e7978 multi_thread dataloader 2019-03-21 14:57:41 +02:00
Glenn Jocher 70fe2204b4 multi_thread dataloader 2019-03-21 14:48:40 +02:00
Glenn Jocher be38caf284 updates 2019-03-21 12:13:09 +02:00
Glenn Jocher 2856af5036 updates 2019-03-21 12:11:08 +02:00
Glenn Jocher aecf840701 updates 2019-03-21 12:09:54 +02:00
Glenn Jocher 03791babfb Merge branch 'master' of /Users/glennjocher/PycharmProjects/yolov3 with conflicts. 2019-03-21 12:08:55 +02:00
Glenn Jocher ad49e70f47
Merge pull request #145 from perry0418/master
Update train.py
2019-03-21 12:04:50 +02:00
Glenn Jocher d661fba8ae updates 2019-03-21 11:48:50 +02:00
perry0418 35396adc9c
Update train.py
solve the Multi-GPU --resume Error #138
https://github.com/ultralytics/yolov3/issues/138
2019-03-21 12:02:57 +08:00
Glenn Jocher 327aaebd7c updates 2019-03-20 22:10:18 +02:00
Glenn Jocher 735b1a370b multi_gpu multi_scale 2019-03-19 15:43:10 +02:00
Glenn Jocher dcdd1ae6b7 multi_gpu multi_scale 2019-03-19 15:35:12 +02:00
Glenn Jocher 76f555c108 multi_gpu multi_scale 2019-03-19 12:34:12 +02:00
Glenn Jocher 056eed2833 multi_gpu multi_scale 2019-03-19 10:38:32 +02:00
Glenn Jocher 45fac6bff1
multi_gpu (#135)
* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates

* updates
2019-03-17 23:45:39 +02:00
Glenn Jocher c1c09eb3cc
Update train.py 2019-03-10 15:03:17 +01:00
Glenn Jocher bc0f30933a updates 2019-03-07 17:16:38 +01:00
Glenn Jocher ff9d343019
Update train.py 2019-03-07 13:28:54 +01:00
Glenn Jocher 6fb14fc903 updates 2019-03-05 17:14:40 +01:00
Glenn Jocher a2ad00d6fc updates 2019-03-05 17:10:34 +01:00
Glenn Jocher e5dc942fee updates 2019-03-04 17:38:38 +01:00
Glenn Jocher 3bea4da604 updates 2019-03-04 17:34:53 +01:00
Glenn Jocher 65ed0d1122 updates 2019-03-04 15:46:25 +01:00
Glenn Jocher 90a20f93e5 updates 2019-02-26 02:53:11 +01:00
Glenn Jocher 9e60e97a6c updates 2019-02-22 16:24:30 +01:00
Glenn Jocher ac22a717f1 updates 2019-02-22 16:15:20 +01:00
Glenn Jocher 12e605165e updates 2019-02-22 15:05:03 +01:00
Glenn Jocher e62736f8a8 updates 2019-02-21 20:16:58 +01:00
Glenn Jocher 485321ecb1 updates 2019-02-21 16:18:11 +01:00
Glenn Jocher af853f604c updates 2019-02-21 15:57:18 +01:00
Glenn Jocher 0b971eddff updates 2019-02-20 18:41:31 +01:00
Glenn Jocher f8c675dbc0 updates 2019-02-20 17:44:41 +01:00
Glenn Jocher a65a383d3a updates 2019-02-20 13:00:39 +01:00
Glenn Jocher 15bba5a345 xy and wh losses respectively merged 2019-02-19 19:55:33 +01:00
Glenn Jocher c828f5459f select GPU0 if multiple available 2019-02-16 14:33:52 +01:00
Glenn Jocher 9706002b71 optimize imports 2019-02-12 18:05:58 +01:00
Glenn Jocher e4e64a9ff6 updates 2019-02-12 13:50:43 +01:00