Glenn Jocher
|
a9dc0c2c29
|
update check_img_size() for model strides
|
5 years ago |
|
Glenn Jocher
|
b9da3ea46a
|
Merge remote-tracking branch 'origin/master'
|
5 years ago |
|
Glenn Jocher
|
3b16c865f0
|
assert --epochs
|
5 years ago |
yxNONG
|
cdb9bde181
|
Unify the check point of single and multi GPU
save the model.hyp etc to checkpoint when use multi GPU training
|
5 years ago |
|
Glenn Jocher
|
bfd51f62f8
|
Merge remote-tracking branch 'origin/master'
|
5 years ago |
|
Glenn Jocher
|
9a9333d245
|
add replicate() to datasets.py
|
5 years ago |
|
Glenn Jocher
|
615d6d0cfa
|
FROM nvcr.io/nvidia/pytorch:20.06-py3
|
5 years ago |
|
Glenn Jocher
|
e8bf385795
|
Merge pull request #223 from ArmyOfRobots/master
Adding torchscript export
|
5 years ago |
|
Glenn Jocher
|
b203c9b7ff
|
update train.py incompatible model message fix #222
|
5 years ago |
|
Glenn Jocher
|
01895f3924
|
Merge pull request #228 from topiaruss/patch-1
Remove non-ascii that breaks pdb single-stepping
|
5 years ago |
Russ Ferriday
|
0d5065d3b9
|
Remove non-ascii that breaks pdb single-stepping
|
5 years ago |
|
Glenn Jocher
|
0bc80e1e57
|
onnx_export.py bug fix #224
|
5 years ago |
|
Glenn Jocher
|
ec8554a39b
|
FROM nvcr.io/nvidia/pytorch:20.06-py3
|
5 years ago |
Jake Poznanski
|
685091a3fc
|
Adding torchscript export
|
5 years ago |
|
Glenn Jocher
|
37e13f8846
|
update mosaic border
|
5 years ago |
|
Glenn Jocher
|
93a6765806
|
update mosaic border
|
5 years ago |
Alex Stoken
|
e18e6811dc
|
Merge branch 'master' into advanced_logging
|
5 years ago |
|
Glenn Jocher
|
77fb8ee082
|
refactor dataloader
|
5 years ago |
|
Glenn Jocher
|
22fb2b0c25
|
refactor dataloader
|
5 years ago |
|
Glenn Jocher
|
97b5186fa0
|
Merge pull request #206 from ml5ah/utils-fix
Fix after suggestion
|
5 years ago |
|
Glenn Jocher
|
8c5b6220f5
|
Update detect.py
|
5 years ago |
|
Glenn Jocher
|
496ec33a33
|
Update detect.py
Added some recent updates that were missing, and updated the filename with an if else.
|
5 years ago |
Anish Hiranandani
|
68f63616b3
|
Fix after suggestion
|
5 years ago |
|
Glenn Jocher
|
256a3e89d2
|
small dataset bug fix #140
|
5 years ago |
|
Glenn Jocher
|
d994ed25f1
|
Merge remote-tracking branch 'origin/master'
|
5 years ago |
|
Glenn Jocher
|
e670a3353b
|
update test.py for FP16 testing during training
|
5 years ago |
|
Glenn Jocher
|
c910ed397c
|
Update README.md
|
5 years ago |
|
Glenn Jocher
|
8669f4561c
|
Update README.md
|
5 years ago |
|
Glenn Jocher
|
8f7e11dcef
|
Update README.md
|
5 years ago |
|
Glenn Jocher
|
b2d4307848
|
Update README.md
|
5 years ago |
|
Glenn Jocher
|
b50fdf16af
|
model.names multi-GPU bug fix #94
|
5 years ago |
|
Glenn Jocher
|
9a9c4f1259
|
update yolo.py default --cfg
|
5 years ago |
|
Alex Stoken
|
de191655e4
|
Fix yaml saving (don't sort keys), reorder --opt keys, bug fix hyp dict accessor
|
5 years ago |
|
Alex Stoken
|
2d396bea00
|
Fix bug in --help from percent sign in help string
|
5 years ago |
|
Glenn Jocher
|
227aa735dc
|
updates
|
5 years ago |
|
Glenn Jocher
|
e071b8dd3a
|
update models/common.py for Conv() flexible padding
|
5 years ago |
|
Glenn Jocher
|
b8557f87e3
|
add stride to datasets.py
|
5 years ago |
|
Glenn Jocher
|
3b062254a6
|
update utils.create_pretrained()
|
5 years ago |
ChristopherSTAN
|
d18adf8df9
|
Add files via upload
|
5 years ago |
|
Glenn Jocher
|
0825cb7fd8
|
create models/hub
|
5 years ago |
|
Glenn Jocher
|
b1b3634145
|
update onnx_export.py #134
|
5 years ago |
|
Glenn Jocher
|
8699c319ca
|
update yolo.py
|
5 years ago |
|
Alex Stoken
|
d1ca6f231d
|
Delete equip_hyp.yaml
|
5 years ago |
|
Alex Stoken
|
611aacf1bf
|
Turn opt into dictionary before sending it to yaml
|
5 years ago |
|
Alex Stoken
|
bc4ef4861b
|
Default optimizer SGD
|
5 years ago |
|
Alex Stoken
|
7abf202cad
|
Mode all optimizer settings to 'hyp.yaml', integrate proper momentum with Adam optimizer
|
5 years ago |
|
Alex Stoken
|
7edbf6570e
|
Fix help message for cfg files
|
5 years ago |
|
Alex Stoken
|
d64ad0fbf3
|
Remove --resume functionality and related checks/logic.
|
5 years ago |
|
Alex Stoken
|
ccf0af18b1
|
Revert coco128.yaml to initial commit
|
5 years ago |
|
Glenn Jocher
|
f9ae460eec
|
update README.md
|
5 years ago |
