85 Commits

Author	SHA1	Message	Date
Glenn Jocher	a586751904	multi-gpu ckpt filesize bug fix #253	2020-07-11 12:35:21 -07:00
Glenn Jocher	5de4e25d68	update tensorboard metric	2020-07-11 11:39:02 -07:00
Glenn Jocher	98fc483abc	train.py results.txt to bucket bug fix	2020-07-11 09:31:53 -07:00
Glenn Jocher	1c13e67b33	evolution bug fix #346	2020-07-09 22:08:42 -07:00
Glenn Jocher	e16e9e43e1	new nc=len(names) check	2020-07-09 17:10:43 -07:00
Glenn Jocher	cb527d3af9	new nc=len(names) check	2020-07-09 17:03:12 -07:00
Glenn Jocher	603ea0bfdc	update log_dir to runs/exp #107	2020-07-09 15:58:07 -07:00
Glenn Jocher	72d5b58b9a	disable LR plot to suppress warning message	2020-07-09 15:16:57 -07:00
Glenn Jocher	24c5a941f0	--resume EMA fix #292	2020-07-09 15:09:06 -07:00
Alex Stoken	9d631408a2	Move hyp and opt yaml save to top of train() ... Fixes bug where scaled values were saved in hyp.yaml, which would cause continuity issues with --resume	2020-07-09 16:18:55 -05:00
Glenn Jocher	bf6f41567a	hyperparameter printout update	2020-07-08 17:21:00 -07:00
Glenn Jocher	dc5e18390a	Merge branch 'master' into advanced_logging	2020-07-08 17:01:19 -07:00
Glenn Jocher	6b134d93c5	Update train.py	2020-07-08 16:58:13 -07:00
Glenn Jocher	16f6834486	update train.py and experimental.py	2020-07-08 14:23:34 -07:00
Alex Stoken	52bac22f09	Add in --resume functionality with option to specify path or to get most recent run	2020-07-07 10:42:28 -05:00
Alex Stoken	f517ba81c7	Merge branch 'master' into advanced_logging	2020-07-06 16:52:11 -05:00
Laughing	956511dafd	fix LR bug	2020-07-05 15:08:24 +08:00
Glenn Jocher	bb3c346916	model.yaml nc inherited from dataset.yaml	2020-07-04 17:51:54 -07:00
Glenn Jocher	df224a0d8f	EMA bug fix #279	2020-07-03 11:56:14 -07:00
Glenn Jocher	3bdea3f697	strip_optimizer() bug fix #253	2020-07-02 21:24:26 -07:00
Glenn Jocher	e02a189a3a	Merge pull request #245 from yxNONG/patch-2 ... Unify the check point of single and multi GPU	2020-07-02 12:05:46 -07:00
Glenn Jocher	597ed4ce63	Update train.py	2020-07-02 12:00:55 -07:00
Glenn Jocher	13f69777a6	typo fix	2020-07-02 09:26:03 -07:00
yxNONG	1aa2b67933	Update train.py	2020-07-02 13:51:52 +08:00
Glenn Jocher	86784cfdbf	--resume bug fix #252	2020-06-30 21:43:53 -07:00
Glenn Jocher	ad4c22cbfe	--resume bug fix #187	2020-06-30 16:16:29 -07:00
Glenn Jocher	3b16c865f0	assert --epochs	2020-06-30 14:08:08 -07:00
yxNONG	cdb9bde181	Unify the check point of single and multi GPU ... save the model.hyp etc to checkpoint when use multi GPU training	2020-06-30 19:06:28 +08:00
Glenn Jocher	b203c9b7ff	update train.py incompatible model message fix #222	2020-06-29 12:45:25 -07:00
Glenn Jocher	37e13f8846	update mosaic border	2020-06-27 13:50:15 -07:00
Alex Stoken	e18e6811dc	Merge branch 'master' into advanced_logging	2020-06-27 10:13:03 -05:00
Glenn Jocher	22fb2b0c25	refactor dataloader	2020-06-26 18:56:13 -07:00
Glenn Jocher	256a3e89d2	small dataset bug fix #140	2020-06-25 17:52:56 -07:00
Glenn Jocher	b50fdf16af	model.names multi-GPU bug fix #94	2020-06-24 22:22:13 -07:00
Alex Stoken	de191655e4	Fix yaml saving (don't sort keys), reorder --opt keys, bug fix hyp dict accessor	2020-06-24 17:21:54 -05:00
Alex Stoken	2d396bea00	Fix bug in --help from percent sign in help string	2020-06-24 16:57:12 -05:00
Glenn Jocher	b8557f87e3	add stride to datasets.py	2020-06-24 13:02:27 -07:00
Alex Stoken	611aacf1bf	Turn opt into dictionary before sending it to yaml	2020-06-24 10:49:08 -05:00
Alex Stoken	bc4ef4861b	Default optimizer SGD	2020-06-24 10:07:43 -05:00
Alex Stoken	7abf202cad	Mode all optimizer settings to 'hyp.yaml', integrate proper momentum with Adam optimizer	2020-06-24 10:03:21 -05:00
Alex Stoken	7edbf6570e	Fix help message for cfg files	2020-06-24 09:45:57 -05:00
Alex Stoken	d64ad0fbf3	Remove --resume functionality and related checks/logic.	2020-06-24 09:17:27 -05:00
Glenn Jocher	6c1b87a42e	update google_utils import	2020-06-22 23:00:23 -07:00
Glenn Jocher	1f1917ef56	remove fast, add merge	2020-06-21 13:37:11 -07:00
Alex Stoken	e572bb0803	Add plot_results save location to log_dir	2020-06-21 09:36:28 -05:00
Lornatang	899f1d4bde	Fix DDP bug in single process multiple device use cases	2020-06-20 13:00:03 +08:00
Glenn Jocher	cdf1eac9f7	Merge pull request #107 from Lornatang/fix-reference-bugs ... fix refrence bug	2020-06-19 14:51:41 -07:00
Glenn Jocher	cce95e744d	backbone as FP16, save default to FP32	2020-06-18 00:13:18 -07:00
Glenn Jocher	d9b64c27c2	save ckpt in FP16 #119	2020-06-17 22:34:13 -07:00
Glenn Jocher	9fdb0fbacf	AutoAnchor bug fix # 117	2020-06-17 19:51:15 -07:00