Glenn Jocher
|
2110f5ee89
|
update log_dir to runs/exp #107
|
4 years ago |
Glenn Jocher
|
603ea0bfdc
|
update log_dir to runs/exp #107
|
4 years ago |
Glenn Jocher
|
72d5b58b9a
|
disable LR plot to suppress warning message
|
4 years ago |
Glenn Jocher
|
24c5a941f0
|
--resume EMA fix #292
|
4 years ago |
Glenn Jocher
|
2b6209a9d5
|
Merge pull request #338 from alexstoken/hyp_save_bugfix
Move hyp and opt yaml save to top of train()
|
4 years ago |
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
|
4 years ago |
Glenn Jocher
|
13a09fa68a
|
Update datasets.py
|
4 years ago |
Glenn Jocher
|
47bf1730fa
|
Update datasets.py
|
4 years ago |
Glenn Jocher
|
f9bab6b127
|
Update datasets.py
|
4 years ago |
lorenzomammana
|
54a9e4f876
|
Refactor code to reduce duplication
|
4 years ago |
lorenzomammana
|
94342acbe1
|
Handle multiple datasets
|
4 years ago |
Glenn Jocher
|
bf6f41567a
|
hyperparameter printout update
|
4 years ago |
Glenn Jocher
|
0fef3f6678
|
Merge pull request #104 from alexstoken/advanced_logging
Log command line options, hyperparameters, and weights per run in `runs/`
|
4 years ago |
Glenn Jocher
|
dc5e18390a
|
Merge branch 'master' into advanced_logging
|
4 years ago |
Glenn Jocher
|
abb024de04
|
Delete new_hyp.yaml
|
4 years ago |
Glenn Jocher
|
6b134d93c5
|
Update train.py
|
4 years ago |
Glenn Jocher
|
22ab1c295f
|
Update test.py
|
4 years ago |
Glenn Jocher
|
c3d3e6b776
|
Update test.py
|
4 years ago |
Glenn Jocher
|
cbe39a1dbb
|
Update utils.py
|
4 years ago |
Glenn Jocher
|
c654d18f00
|
Update utils.py
|
4 years ago |
Glenn Jocher
|
16f6834486
|
update train.py and experimental.py
|
4 years ago |
Glenn Jocher
|
1b9e28e72e
|
Update requirements.txt
|
4 years ago |
Glenn Jocher
|
31b07b3550
|
Merge pull request #275 from wanghaoyang0106/master
[bug fix] potential problem if img fed to model is in rectangular shape
|
4 years ago |
Glenn Jocher
|
cd6609297c
|
Merge pull request #321 from dlawrences/master
Update CoreML model conversion to do pixel scaling as well
|
4 years ago |
Laurentiu Diaconu
|
7c4d3cf51b
|
Merge pull request #1 from dlawrences/dev/coreml_convert_pxscaling
Updated coreml conversion to do pixel scaling between 0.0 and 1.0
|
4 years ago |
Laurentiu Diaconu
|
a9918fbf2b
|
updated coreml conversion to do pixel scaling between 0.0 and 1.0
|
4 years ago |
Glenn Jocher
|
e8cf24b6c8
|
Initial model ensemble capability #318
|
4 years ago |
Alex Stoken
|
e6e7e7f1ac
|
Delete advanced_logging_test.ipynb
Remove from advanced_logging branch before merge.
|
4 years ago |
Alex Stoken
|
43647514da
|
Colab example/test of new features
|
4 years ago |
Alex Stoken
|
95f0a56df7
|
Bug fix to get_latest_run when recent run is named
|
4 years ago |
Alex Stoken
|
52bac22f09
|
Add in --resume functionality with option to specify path or to get most recent run
|
4 years ago |
Alex Stoken
|
8b6dbb7cfc
|
Add optimizer choice to hyp file
|
4 years ago |
Alex Stoken
|
5ac517b22a
|
Remove duplicate `verbose` arg in test.py
|
4 years ago |
Alex Stoken
|
f517ba81c7
|
Merge branch 'master' into advanced_logging
|
4 years ago |
Glenn Jocher
|
121d90b3f2
|
update fuse_conv_and_bn()
|
4 years ago |
Glenn Jocher
|
6b95d6d2c0
|
.to(device) bug fix
|
4 years ago |
Glenn Jocher
|
a62333e808
|
test.py .fuse() update
|
4 years ago |
Glenn Jocher
|
a40f615c6f
|
.half() bug fix
|
4 years ago |
Glenn Jocher
|
04bdbe4104
|
fuse update
|
4 years ago |
Glenn Jocher
|
5ba1de0cdc
|
update experimental.py with Ensemble() module
|
4 years ago |
Glenn Jocher
|
38f5c1ad1d
|
pruning and sparsity initial commit
|
4 years ago |
Glenn Jocher
|
997ba7b346
|
import yaml in yolo.py
|
4 years ago |
Glenn Jocher
|
8c43a6906d
|
Update README.md
|
4 years ago |
Glenn Jocher
|
2541f77946
|
update detect.py
|
4 years ago |
Glenn Jocher
|
b7fe1d0622
|
Merge pull request #300 from Laughing-q/master
fix LR bug
|
4 years ago |
Laughing
|
956511dafd
|
fix LR bug
|
4 years ago |
Glenn Jocher
|
762b06de4b
|
Merge remote-tracking branch 'origin/master'
|
4 years ago |
Glenn Jocher
|
4e2d24602d
|
update yolo.py
|
4 years ago |
Glenn Jocher
|
bb3c346916
|
model.yaml nc inherited from dataset.yaml
|
4 years ago |
Glenn Jocher
|
1fdaa4987b
|
update detect.py
|
4 years ago |