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 |
|
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 |
|
Alex Stoken
|
d34291733b
|
Fix get_latest_run() to search inside 'weights' subfolders
|
2020-06-17 10:26:55 -05:00 |
|
Alex Stoken
|
5f2eeba233
|
remove old print statements
|
2020-06-16 17:09:39 -05:00 |
|
Alex Stoken
|
333f678b37
|
add update default hyp dict with provided yaml
|
2020-06-16 16:36:20 -05:00 |
|
Alex Stoken
|
a448c3bcd7
|
add logic for resuming and getting hyp for resume run
|
2020-06-16 16:30:12 -05:00 |
|
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
|
2020-06-16 15:50:27 -05:00 |
|
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
|
2020-06-16 15:13:03 -05:00 |
|
Alex Stoken
|
4418809cf5
|
change weights dir (wdir) to be unique to each run, under log_dir
|
2020-06-16 15:09:51 -05:00 |
|
Alex Stoken
|
d9f446cd81
|
add save yaml of opt and hyp to tensorboard log_dir in train()
|
2020-06-16 15:06:13 -05:00 |
|
Alex Stoken
|
a85e6d0fc0
|
add parser arg for hyp yaml file
|
2020-06-16 14:53:32 -05:00 |
Glenn Jocher
|
8b26e89006
|
AutoAnchor bug fix #72
|
2020-06-16 00:53:34 -07:00 |
|
Glenn Jocher
|
8fa3724072
|
kmean_anchors() update
|
2020-06-16 00:40:21 -07:00 |
|
Glenn Jocher
|
bdd9fee841
|
update fast mode
|
2020-06-16 00:29:54 -07:00 |
|
Glenn Jocher
|
1c0b6236e3
|
update fast mode
|
2020-06-16 00:11:29 -07:00 |
|
Glenn Jocher
|
db2c3acd3a
|
updated testing settings, rebalanced towards FP16 latency
|
2020-06-15 16:55:57 -07:00 |
|
Glenn Jocher
|
915b1481fc
|
default check_git_status() to True
|
2020-06-15 16:18:46 -07:00 |
|
Glenn Jocher
|
0b514daced
|
FP16 test loss bug fix
|
2020-06-15 15:26:29 -07:00 |
|
Glenn Jocher
|
a1748a8d6e
|
test during training default to FP16
|
2020-06-15 15:20:27 -07:00 |
|
Glenn Jocher
|
bd3e389130
|
update google_utils.py
|
2020-06-15 14:39:04 -07:00 |
|
Glenn Jocher
|
76ca367a01
|
FP16 to FP32 ckpt load
|
2020-06-15 13:32:45 -07:00 |
|
Glenn Jocher
|
6fb5ff014c
|
FP16 to FP32 ckpt load
|
2020-06-15 13:21:34 -07:00 |
|
Glenn Jocher
|
14523bb030
|
FP16 to FP32 ckpt load
|
2020-06-15 13:18:39 -07:00 |
|
Glenn Jocher
|
c5966abba8
|
glob search bug fix #77
|
2020-06-15 12:08:57 -07:00 |
|
Glenn Jocher
|
ace56eb5b6
|
Merge pull request #70 from Lornatang/fix-reference-bugs
fix reference problem
|
2020-06-15 11:18:17 -07:00 |
Lornatang
|
54795a4f0a
|
fix problem
|
2020-06-15 16:11:06 +08:00 |
|
Glenn Jocher
|
45a17ea40c
|
Merge remote-tracking branch 'origin/master'
|
2020-06-15 00:44:08 -07:00 |
|
Glenn Jocher
|
e7c24e04ac
|
half() fix #66
|
2020-06-15 00:43:57 -07:00 |
|
Glenn Jocher
|
6570336284
|
Merge pull request #65 from Ashing00/master
add "__init__.py" file in <models> folder
|
2020-06-15 00:34:54 -07:00 |
ashing
|
62eda5e9c2
|
add "__init__.py" file in <models> folder
|
2020-06-15 14:37:07 +08:00 |
|
Glenn Jocher
|
111994956e
|
switch default inference to FP16 on GPU
|
2020-06-14 19:25:39 -07:00 |
|
Glenn Jocher
|
157aff2854
|
FP16 inference fix
|
2020-06-14 18:45:20 -07:00 |
|
Glenn Jocher
|
c6fc4242a2
|
Merge remote-tracking branch 'origin/master'
|
2020-06-14 18:32:35 -07:00 |
|
Glenn Jocher
|
260b1729f0
|
FP16 inference update
|
2020-06-14 18:32:29 -07:00 |
|
Glenn Jocher
|
b545ee32f4
|
Created using Colaboratory
|
2020-06-14 13:29:05 -07:00 |
|
Glenn Jocher
|
629d3701fc
|
pyyaml >= 5.3
|
2020-06-14 12:42:51 -07:00 |
|
Glenn Jocher
|
4b0e3745f2
|
Merge pull request #59 from Lijun-Yu/master
Fix config path for PyTorch Hub
|
2020-06-14 12:16:41 -07:00 |
|
Glenn Jocher
|
4f6c0cf4c5
|
Update hubconf.py
|
2020-06-14 12:15:14 -07:00 |
Lijun Yu
|
0dec8ffd66
|
path fix for torch hub
|
2020-06-14 14:47:40 -04:00 |
|
Glenn Jocher
|
4dd75b3635
|
pycocotools git install fix
|
2020-06-14 10:54:00 -07:00 |
|
Glenn Jocher
|
9bf35e7b86
|
Merge pull request #56 from channeladam/master
Fix initial basic setup / new starter issues
|
2020-06-14 10:33:34 -07:00 |
|
Glenn Jocher
|
335c607a24
|
Update datasets.py
|
2020-06-14 10:32:59 -07:00 |
|
Glenn Jocher
|
1374581929
|
Update requirements.txt
|
2020-06-14 10:23:33 -07:00 |
Adam
|
f47ca2085c
|
Include yolov5x extra large model weights
|
2020-06-14 18:19:23 +10:00 |
|
Adam
|
34bf936ce6
|
Add more logging for diagnostics
|
2020-06-14 18:18:57 +10:00 |
|
Adam
|
faa6aeaa20
|
Fix pip requirements issues
|
2020-06-14 18:18:31 +10:00 |
|
Glenn Jocher
|
93bbf1b87a
|
best possible recall fix
|
2020-06-13 15:25:53 -07:00 |
|
Glenn Jocher
|
31f3310029
|
assert best possible recall > 0.9 before training
|
2020-06-13 15:05:41 -07:00 |