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 |
|
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 |
|
Glenn Jocher
|
fc171e2678
|
check_anchor_order() update
|
2020-06-21 14:08:59 -07:00 |
|
Glenn Jocher
|
1f1917ef56
|
remove fast, add merge
|
2020-06-21 13:37:11 -07:00 |
|
Alex Stoken
|
e572bb0803
|
Add plot_results save location to log_dir
|
2020-06-21 09:36:28 -05:00 |
|
Glenn Jocher
|
24dd150fbd
|
update torch>=1.4
|
2020-06-20 11:55:39 -07:00 |
|
Glenn Jocher
|
ba340ed72d
|
Merge pull request #143 from Lornatang/Fix-multi-GPU-training-errors
Fix DDP bug in single process multiple device use cases
|
2020-06-20 11:10:40 -07:00 |
|
Glenn Jocher
|
024a42f4fd
|
datasets.py vid_formats updated
|
2020-06-20 09:53:37 -07:00 |
|
Glenn Jocher
|
5c2cd711df
|
vid_formats bug fix #146
|
2020-06-20 09:26:07 -07:00 |
Lornatang
|
899f1d4bde
|
Fix DDP bug in single process multiple device use cases
|
2020-06-20 13:00:03 +08:00 |
|
Glenn Jocher
|
0a09882e4f
|
Update issue templates
|
2020-06-19 16:46:21 -07:00 |
|
Glenn Jocher
|
cdf1eac9f7
|
Merge pull request #107 from Lornatang/fix-reference-bugs
fix refrence bug
|
2020-06-19 14:51:41 -07:00 |
|
Glenn Jocher
|
458c4d0e0b
|
Update README.md
|
2020-06-19 12:14:06 -07:00 |
|
Glenn Jocher
|
34ec87ecf9
|
Update README.md
|
2020-06-19 12:13:15 -07:00 |
|
Glenn Jocher
|
d4c6674c98
|
speeds updated from FP32 to FP16
|
2020-06-19 12:05:10 -07:00 |
|
Glenn Jocher
|
e249792c2f
|
update onnx_export.py usage
|
2020-06-19 10:15:08 -07:00 |
|
Glenn Jocher
|
80b82e8bde
|
update yolo.py
|
2020-06-18 19:33:02 -07:00 |
|
Glenn Jocher
|
245bd1d242
|
update README.md
|
2020-06-18 12:46:30 -07:00 |
|
Glenn Jocher
|
a3ee0a8954
|
check_img_size() update
|
2020-06-18 11:46:04 -07:00 |
|
Glenn Jocher
|
10145833c3
|
strip_optimizer() default to save FP16 model
|
2020-06-18 00:16:47 -07:00 |
|
Glenn Jocher
|
cce95e744d
|
backbone as FP16, save default to FP32
|
2020-06-18 00:13:18 -07:00 |
|
Glenn Jocher
|
d9b64c27c2
|
save ckpt in FP16 #119
|
2020-06-17 22:34:13 -07:00 |
|
Glenn Jocher
|
9fdb0fbacf
|
AutoAnchor bug fix # 117
|
2020-06-17 19:51:15 -07:00 |
|
Alex Stoken
|
c8152c81a6
|
Syntax fixes
|
2020-06-17 16:32:13 -05:00 |
|
Alex Stoken
|
9b7386f603
|
Add save_dir arg to test.test, use arg as location for saving batch jpgs
|
2020-06-17 16:08:46 -05:00 |
|
Alex Stoken
|
945307beba
|
Add save_dir to plot_lr_scheduler and plot_labels
Set save_dir = log_dir in train.py
|
2020-06-17 16:03:18 -05:00 |
|
Alex Stoken
|
3b2b330872
|
Move results.txt from weights/ to log_dir
|
2020-06-17 15:55:45 -05:00 |
|
Glenn Jocher
|
8b6f5826bc
|
update README.md
|
2020-06-17 12:58:01 -07:00 |
|
Glenn Jocher
|
2bfd242fa1
|
Merge remote-tracking branch 'origin/master'
|
2020-06-17 12:57:03 -07:00 |
|
Glenn Jocher
|
b7ac446d62
|
update README.md
|
2020-06-17 12:56:47 -07:00 |
|
Alex Stoken
|
ade023cff2
|
Fix hyp file read in and dict update.
Add example of hyp yaml
|
2020-06-17 10:59:20 -05:00 |
|
Alex Stoken
|
3263a204ea
|
Fix get_latest_run to search 'search_dir' recursivly
|
2020-06-17 10:34:37 -05:00 |