Glenn Jocher
fc171e2678
check_anchor_order() update
4 years ago
Glenn Jocher
1f1917ef56
remove fast, add merge
4 years ago
Alex Stoken
e572bb0803
Add plot_results save location to log_dir
4 years ago
Glenn Jocher
24dd150fbd
update torch>=1.4
4 years ago
Glenn Jocher
ba340ed72d
Merge pull request #143 from Lornatang/Fix-multi-GPU-training-errors
Fix DDP bug in single process multiple device use cases
4 years ago
Glenn Jocher
024a42f4fd
datasets.py vid_formats updated
4 years ago
Glenn Jocher
5c2cd711df
vid_formats bug fix #146
4 years ago
Lornatang
899f1d4bde
Fix DDP bug in single process multiple device use cases
4 years ago
Glenn Jocher
0a09882e4f
Update issue templates
4 years ago
Glenn Jocher
cdf1eac9f7
Merge pull request #107 from Lornatang/fix-reference-bugs
fix refrence bug
4 years ago
Glenn Jocher
458c4d0e0b
Update README.md
4 years ago
Glenn Jocher
34ec87ecf9
Update README.md
4 years ago
Glenn Jocher
d4c6674c98
speeds updated from FP32 to FP16
4 years ago
Glenn Jocher
e249792c2f
update onnx_export.py usage
4 years ago
Glenn Jocher
80b82e8bde
update yolo.py
4 years ago
Glenn Jocher
245bd1d242
update README.md
4 years ago
Glenn Jocher
a3ee0a8954
check_img_size() update
4 years ago
Glenn Jocher
10145833c3
strip_optimizer() default to save FP16 model
4 years ago
Glenn Jocher
cce95e744d
backbone as FP16, save default to FP32
4 years ago
Glenn Jocher
d9b64c27c2
save ckpt in FP16 #119
4 years ago
Glenn Jocher
9fdb0fbacf
AutoAnchor bug fix # 117
4 years ago
Alex Stoken
c8152c81a6
Syntax fixes
4 years ago
Alex Stoken
9b7386f603
Add save_dir arg to test.test, use arg as location for saving batch jpgs
4 years ago
Alex Stoken
945307beba
Add save_dir to plot_lr_scheduler and plot_labels
Set save_dir = log_dir in train.py
4 years ago
Alex Stoken
3b2b330872
Move results.txt from weights/ to log_dir
4 years ago
Glenn Jocher
8b6f5826bc
update README.md
4 years ago
Glenn Jocher
2bfd242fa1
Merge remote-tracking branch 'origin/master'
4 years ago
Glenn Jocher
b7ac446d62
update README.md
4 years ago
Alex Stoken
ade023cff2
Fix hyp file read in and dict update.
Add example of hyp yaml
4 years ago
Alex Stoken
3263a204ea
Fix get_latest_run to search 'search_dir' recursivly
4 years ago
Alex Stoken
d34291733b
Fix get_latest_run() to search inside 'weights' subfolders
4 years ago
Glenn Jocher
91de3381e1
Merge pull request #112 from imprashr/patch-1
onnx should be 1.6.0
4 years ago
Prashant Rai
0071b69994
onnx should be 1.6.0
for smoothly export model to onnx, onnx should be 1.6.0
4 years ago
Lornatang
2368603484
fix refrence bug
In torch==1.5, the import of the API has changed. Although it does not interrupt the operation of the program, it seems to me to be an implicit error and may throw an exception in later versions.
4 years ago
Glenn Jocher
a557b7d924
check_img_size() update
4 years ago
Glenn Jocher
034609414e
AutoAnchor fitness to screen
4 years ago
Glenn Jocher
ef75f7313d
AutoAnchor n, imgsz bug fix
4 years ago
Glenn Jocher
2daa412575
train with multi-gpu half test bug fix #99
4 years ago
Glenn Jocher
d1e5716610
train with multi-gpu half test bug fix #99
4 years ago
Glenn Jocher
1dfc28527f
AutoAnchor BPR to 4 significant figures
4 years ago
Glenn Jocher
8db51c7002
tb_writer bug fix
4 years ago
Glenn Jocher
4052603e1f
AutoAnchor update - improvement check
4 years ago
Alex Stoken
5f2eeba233
remove old print statements
4 years ago
Glenn Jocher
afe1df385b
dist.destroy_process_group() bug fix
4 years ago
Glenn Jocher
72d06147b2
Multi-GPU disabled in test.py
4 years ago
Alex Stoken
333f678b37
add update default hyp dict with provided yaml
4 years ago
Alex Stoken
a448c3bcd7
add logic for resuming and getting hyp for resume run
4 years ago
Glenn Jocher
bafbc65ee3
AutoAnchor bug fix
4 years ago
Alex Stoken
25e51bcec7
add util function to get most recent last.pt file
added logic in train.py __main__ to handle resuming from a run
4 years ago
Alex Stoken
490f1e7b9c
add save_dir arg to plot_lr_scheduler, default to current dir.
Uncomment plot_lr_scheduler in train() and pass log_dir as save location
4 years ago