Commit Graph

2489 Commits (bfa1f23045c7c4136a9b8ced9d6be8249ed72692)
 

Author SHA1 Message Date
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
Alex Stoken e18e6811dc
Merge branch 'master' into advanced_logging 2020-06-27 10:13:03 -05: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 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
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 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
ChristopherSTAN d18adf8df9
Add files via upload 2020-06-24 15:53:34 -04: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
Alex Stoken d1ca6f231d
Delete equip_hyp.yaml 2020-06-24 11:22:12 -05: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
Alex Stoken ccf0af18b1 Revert coco128.yaml to initial commit 2020-06-24 09:13:41 -05: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