Commit Graph

748 Commits

Author SHA1 Message Date
Timothy M. Shead ada2958105
Fix argparse string escapes in train.py. (#1045) 2020-04-12 10:00:50 -07:00
Glenn Jocher 6e19245dc8 auto strip optimizer from best.pt after training 2020-04-09 19:53:29 -07:00
Glenn Jocher d1601ae0f3 training updates 2020-04-08 21:34:34 -07:00
Glenn Jocher 4120ac3aa6 training updates 2020-04-08 21:01:58 -07:00
Glenn Jocher 97780cfdb4 parameterize grid size 2020-04-08 10:14:33 -07:00
Glenn Jocher 4fa3fd2df3 tensorboard notice 2020-04-06 16:33:23 -07:00
Glenn Jocher 05ae6e8499 tensorboard/focal loss reporting update 2020-04-06 15:45:18 -07:00
Glenn Jocher c7f93bae40 update coco-tuned hyp['cls'] to current dataset 2020-04-06 10:58:07 -07:00
Glenn Jocher e81a152a92 tensorboard notice and model verbose option 2020-04-05 13:35:58 -07:00
Glenn Jocher 41246aa042 tensorboard updates 2020-04-04 19:34:39 -07:00
Glenn Jocher 9155ef3f4f burnin merged with prebias 2020-04-02 14:08:21 -07:00
Glenn Jocher 992e0d7cb4 default test --conf to 0.001 2020-03-31 14:36:25 -07:00
Glenn Jocher b2d9f1898f burnin lr ramp 300 iterations 2020-03-30 19:27:42 -07:00
Glenn Jocher eb151a881e NMS and test batch_size updates 2020-03-29 20:41:32 -07:00
Glenn Jocher c6b59a0e8a LR schedule to 0.05 min 2020-03-29 13:29:06 -07:00
Glenn Jocher 9c5e76b93d EMA implemented by default 2020-03-29 13:14:54 -07:00
Yonghye Kwon 20b2671de0
cleanup (#963)
* cleanup

cleanup

* Update train.py

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2020-03-26 14:46:17 -07:00
Glenn Jocher 448c4a6e1f Remove deprecated --arc architecture options, implement --arc default for all cases 2020-03-16 20:46:25 -07:00
Glenn Jocher 851c9b9883 EMA class updates 2020-03-14 17:34:13 -07:00
Glenn Jocher d91469a516 EMA class updates 2020-03-14 17:33:29 -07:00
Glenn Jocher 9ce4ec48a7 model.info() method implemented 2020-03-14 16:46:54 -07:00
Glenn Jocher b89cc396af EMA class updates 2020-03-14 16:23:14 -07:00
Glenn Jocher a52c0abf8d updates 2020-03-13 20:12:54 -07:00
Glenn Jocher 418269d739 FocalLoss() gamma and alpha default values 2020-03-13 16:51:30 -07:00
Glenn Jocher 208b9a73fe updates 2020-03-13 16:08:49 -07:00
Glenn Jocher f30a8706e5 update to coco results82 2020-03-13 12:06:48 -07:00
Glenn Jocher 41bf46a419 updates 2020-03-11 22:11:19 -07:00
Glenn Jocher 75e88561cb updates 2020-03-11 20:45:14 -07:00
Glenn Jocher e76d4d0ffc updates 2020-03-11 17:13:40 -07:00
Glenn Jocher e40d4c87f2 updates 2020-03-11 15:57:37 -07:00
Glenn Jocher 4089735c5e updates 2020-03-11 14:50:50 -07:00
Glenn Jocher 320f9c6601 updates 2020-03-11 12:18:03 -07:00
Glenn Jocher 7a83574022 updates 2020-03-10 12:17:23 -07:00
Glenn Jocher 378f08c6d5 updates 2020-03-05 12:30:11 -08:00
Glenn Jocher 1dc1761f45 updates 2020-03-05 10:20:08 -08:00
Glenn Jocher 8b6c8a5318 updates 2020-03-04 16:33:14 -08:00
Glenn Jocher 305c07bac8 updates 2020-03-04 13:24:18 -08:00
Glenn Jocher 981b452b1d updates 2020-03-04 13:20:08 -08:00
Glenn Jocher 6ab753a9e7 updates 2020-03-04 13:06:31 -08:00
Glenn Jocher dce753ead4 updates 2020-03-02 14:30:01 -08:00
Glenn Jocher 84371f6811 updates 2020-03-01 21:33:16 -08:00
Glenn Jocher 7823473d2f updates 2020-03-01 20:55:20 -08:00
Glenn Jocher d5815ebfd2 updates 2020-02-27 13:40:14 -08:00
Glenn Jocher 7d7c22cb7e updates 2020-02-26 13:42:50 -08:00
Glenn Jocher f743235fac updates 2020-02-24 12:44:22 -08:00
Glenn Jocher 24957dca98 updates 2020-02-24 12:21:47 -08:00
Glenn Jocher ef3bd7e12b updates 2020-02-24 09:06:17 -08:00
Glenn Jocher 2624d55623 updates 2020-02-22 21:24:56 -08:00
Glenn Jocher b052085cc4 updates 2020-02-22 21:21:45 -08:00
Glenn Jocher 817c0bfeed updates 2020-02-22 21:17:38 -08:00