yxNONG
|
53cdaf6bf5
|
Update utils.py
|
2020-07-02 13:48:19 +08:00 |
Glenn Jocher
|
a03adb5ab8
|
Merge remote-tracking branch 'origin/master'
|
2020-07-01 19:16:06 -07:00 |
|
Glenn Jocher
|
1fca7a7f24
|
autopad() update in common.py
|
2020-07-01 19:15:59 -07:00 |
|
Glenn Jocher
|
5323ad224d
|
export.py update
|
2020-07-01 16:15:25 -07:00 |
|
Glenn Jocher
|
a62a1c2c67
|
export.py update
|
2020-07-01 16:14:49 -07:00 |
|
Glenn Jocher
|
f1d67f4110
|
update export.py
|
2020-07-01 15:46:15 -07:00 |
|
Glenn Jocher
|
e74ccb2985
|
Merge remote-tracking branch 'origin/master'
|
2020-07-01 11:44:55 -07:00 |
|
Glenn Jocher
|
b5659d1195
|
module updates
|
2020-07-01 11:44:49 -07:00 |
|
Glenn Jocher
|
dfd63de20a
|
Merge pull request #255 from edurenye/ignore_torchscript
Add torchscript files to gitignore
|
2020-07-01 10:10:11 -07:00 |
edurenye
|
1c86c2cb66
|
Add torchscript files to gitignore
|
2020-07-01 10:04:26 +02:00 |
|
Glenn Jocher
|
86784cfdbf
|
--resume bug fix #252
|
2020-06-30 21:43:53 -07:00 |
|
Glenn Jocher
|
0a08375a8a
|
Merge remote-tracking branch 'origin/master'
|
2020-06-30 16:42:16 -07:00 |
|
Glenn Jocher
|
4e62eb90be
|
check_img_size() int() update
|
2020-06-30 16:41:26 -07:00 |
|
Glenn Jocher
|
5c1c73ecda
|
Merge remote-tracking branch 'origin/master'
|
2020-06-30 16:16:36 -07:00 |
|
Glenn Jocher
|
ad4c22cbfe
|
--resume bug fix #187
|
2020-06-30 16:16:29 -07:00 |
|
Glenn Jocher
|
2b7a11300e
|
Merge remote-tracking branch 'origin/master'
|
2020-06-30 15:25:45 -07:00 |
|
Glenn Jocher
|
a9dc0c2c29
|
update check_img_size() for model strides
|
2020-06-30 15:25:38 -07:00 |
|
Glenn Jocher
|
b9da3ea46a
|
Merge remote-tracking branch 'origin/master'
|
2020-06-30 14:08:14 -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
|
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 |