214 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
yxNONG	53cdaf6bf5	Update utils.py	2020-07-02 13:48:19 +08: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	bfd51f62f8	Merge remote-tracking branch 'origin/master'	2020-06-29 17:10:41 -07:00
Glenn Jocher	9a9333d245	add replicate() to datasets.py	2020-06-29 17:10:33 -07:00
Glenn Jocher	615d6d0cfa	FROM nvcr.io/nvidia/pytorch:20.06-py3	2020-06-29 14:00:13 -07:00
Glenn Jocher	e8bf385795	Merge pull request #223 from ArmyOfRobots/master ... Adding torchscript export	2020-06-29 13:39:53 -07:00
Glenn Jocher	b203c9b7ff	update train.py incompatible model message fix #222	2020-06-29 12:45:25 -07:00
Glenn Jocher	01895f3924	Merge pull request #228 from topiaruss/patch-1 ... Remove non-ascii that breaks pdb single-stepping	2020-06-29 10:08:02 -07:00
Russ Ferriday	0d5065d3b9	Remove non-ascii that breaks pdb single-stepping	2020-06-29 12:56:40 +01:00
Glenn Jocher	0bc80e1e57	onnx_export.py bug fix #224	2020-06-28 17:07:03 -07:00
Glenn Jocher	ec8554a39b	FROM nvcr.io/nvidia/pytorch:20.06-py3	2020-06-28 13:26:31 -07:00
Jake Poznanski	685091a3fc	Adding torchscript export	2020-06-28 13:17:05 -07:00
Glenn Jocher	37e13f8846	update mosaic border	2020-06-27 13:50:15 -07:00
Glenn Jocher	93a6765806	update mosaic border	2020-06-27 13:02:01 -07:00
Glenn Jocher	77fb8ee082	refactor dataloader	2020-06-26 19:30:09 -07:00
Glenn Jocher	22fb2b0c25	refactor dataloader	2020-06-26 18:56:13 -07:00
Glenn Jocher	97b5186fa0	Merge pull request #206 from ml5ah/utils-fix ... Fix after suggestion	2020-06-25 19:26:50 -07:00
Glenn Jocher	8c5b6220f5	Update detect.py	2020-06-25 19:26:23 -07:00
Glenn Jocher	496ec33a33	Update detect.py ... Added some recent updates that were missing, and updated the filename with an if else.	2020-06-25 19:19:15 -07:00
Anish Hiranandani	68f63616b3	Fix after suggestion	2020-06-25 18:23:36 -07:00
Glenn Jocher	256a3e89d2	small dataset bug fix #140	2020-06-25 17:52:56 -07:00
Glenn Jocher	d994ed25f1	Merge remote-tracking branch 'origin/master'	2020-06-25 15:09:12 -07:00
Glenn Jocher	e670a3353b	update test.py for FP16 testing during training	2020-06-25 15:09:04 -07:00
Glenn Jocher	c910ed397c	Update README.md	2020-06-25 14:27:45 -07:00
Glenn Jocher	8669f4561c	Update README.md	2020-06-25 12:26:30 -07:00
Glenn Jocher	8f7e11dcef	Update README.md v1.0	2020-06-25 11:53:13 -07:00
Glenn Jocher	b2d4307848	Update README.md	2020-06-25 10:59:42 -07:00
Glenn Jocher	b50fdf16af	model.names multi-GPU bug fix #94	2020-06-24 22:22:13 -07:00
Glenn Jocher	9a9c4f1259	update yolo.py default --cfg	2020-06-24 18:00:43 -07:00
Glenn Jocher	227aa735dc	updates	2020-06-24 13:20:14 -07:00
Glenn Jocher	e071b8dd3a	update models/common.py for Conv() flexible padding	2020-06-24 13:12:23 -07:00
Glenn Jocher	b8557f87e3	add stride to datasets.py	2020-06-24 13:02:27 -07:00
Glenn Jocher	3b062254a6	update utils.create_pretrained()	2020-06-24 12:53:43 -07:00
Glenn Jocher	0825cb7fd8	create models/hub	2020-06-24 11:58:23 -07:00
Glenn Jocher	b1b3634145	update onnx_export.py #134	2020-06-24 11:00:03 -07:00
Glenn Jocher	8699c319ca	update yolo.py	2020-06-24 10:42:01 -07:00
Glenn Jocher	f9ae460eec	update README.md	2020-06-23 20:12:23 -07:00
Glenn Jocher	ec72eea62b	Merge remote-tracking branch 'origin/master'	2020-06-23 10:53:48 -07:00
Glenn Jocher	53294f28b4	update yolo.py	2020-06-23 10:53:42 -07:00
Glenn Jocher	6cea1bff60	Update README.md	2020-06-23 10:11:34 -07:00
Glenn Jocher	a63e1c9b3c	study update	2020-06-22 23:23:50 -07:00
Glenn Jocher	2a550333c0	update README.md	2020-06-22 23:20:37 -07:00
Glenn Jocher	db898131f9	study update	2020-06-22 23:07:22 -07:00
Glenn Jocher	6c1b87a42e	update google_utils import	2020-06-22 23:00:23 -07:00
Glenn Jocher	7306dadf78	Update README.md	2020-06-22 15:16:10 -07:00
Glenn Jocher	3804578b78	Update README.md	2020-06-22 14:33:47 -07:00
Glenn Jocher	69ff486693	Update README.md	2020-06-22 14:32:05 -07:00
Glenn Jocher	364fcfd7db	PANet update	2020-06-22 14:27:17 -07:00
Glenn Jocher	ef58dac963	update Dockerfile	2020-06-22 10:24:56 -07:00
Glenn Jocher	bbd12c7cfe	detect.py fast bug fix	2020-06-22 10:14:58 -07:00