Glenn Jocher
|
140d84cca1
|
comment updates
|
2020-07-13 12:17:52 -07:00 |
|
Glenn Jocher
|
41bb70b325
|
update create_pretrained()
|
2020-07-10 00:25:35 -07:00 |
|
Glenn Jocher
|
1531795f5e
|
avoid git status check in Docker bug fix #342
|
2020-07-10 00:11:50 -07:00 |
|
Glenn Jocher
|
1e9cf6a6e9
|
add file sizes to ckpt ops
|
2020-07-10 00:00:19 -07:00 |
|
Glenn Jocher
|
2110f5ee89
|
update log_dir to runs/exp #107
|
2020-07-09 16:03:10 -07:00 |
|
Glenn Jocher
|
603ea0bfdc
|
update log_dir to runs/exp #107
|
2020-07-09 15:58:07 -07:00 |
|
Glenn Jocher
|
bf6f41567a
|
hyperparameter printout update
|
2020-07-08 17:21:00 -07:00 |
|
Glenn Jocher
|
dc5e18390a
|
Merge branch 'master' into advanced_logging
|
2020-07-08 17:01:19 -07:00 |
|
Glenn Jocher
|
cbe39a1dbb
|
Update utils.py
|
2020-07-08 16:14:32 -07:00 |
|
Glenn Jocher
|
c654d18f00
|
Update utils.py
|
2020-07-08 16:11:22 -07:00 |
|
Glenn Jocher
|
31b07b3550
|
Merge pull request #275 from wanghaoyang0106/master
[bug fix] potential problem if img fed to model is in rectangular shape
|
2020-07-08 00:07:54 -07:00 |
Alex Stoken
|
95f0a56df7
|
Bug fix to get_latest_run when recent run is named
|
2020-07-07 11:19:49 -05:00 |
|
Alex Stoken
|
f517ba81c7
|
Merge branch 'master' into advanced_logging
|
2020-07-06 16:52:11 -05:00 |
|
Glenn Jocher
|
ce0c58f678
|
update compute_loss()
|
2020-07-03 18:56:07 -07:00 |
|
Glenn Jocher
|
f767023c56
|
offset and balance update
|
2020-07-03 14:07:00 -07:00 |
wanghaoyang0106
|
b57f83d005
|
[bug fix] potential problem if img fed to model is in rectangular shape
|
2020-07-03 13:09:21 +08:00 |
|
Glenn Jocher
|
4e62eb90be
|
check_img_size() int() update
|
2020-06-30 16:41:26 -07:00 |
|
Alex Stoken
|
e18e6811dc
|
Merge branch 'master' into advanced_logging
|
2020-06-27 10:13:03 -05:00 |
|
Glenn Jocher
|
3b062254a6
|
update utils.create_pretrained()
|
2020-06-24 12:53:43 -07:00 |
|
Glenn Jocher
|
6c1b87a42e
|
update google_utils import
|
2020-06-22 23:00:23 -07:00 |
|
Glenn Jocher
|
364fcfd7db
|
PANet update
|
2020-06-22 14:27:17 -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
|
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
|
9fdb0fbacf
|
AutoAnchor bug fix # 117
|
2020-06-17 19:51:15 -07: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
|
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 |
|
Glenn Jocher
|
a557b7d924
|
check_img_size() update
|
2020-06-16 17:48:01 -07:00 |
|
Glenn Jocher
|
034609414e
|
AutoAnchor fitness to screen
|
2020-06-16 17:08:45 -07:00 |
|
Glenn Jocher
|
ef75f7313d
|
AutoAnchor n, imgsz bug fix
|
2020-06-16 16:48:01 -07:00 |
|
Glenn Jocher
|
1dfc28527f
|
AutoAnchor BPR to 4 significant figures
|
2020-06-16 16:19:58 -07:00 |
|
Glenn Jocher
|
4052603e1f
|
AutoAnchor update - improvement check
|
2020-06-16 15:54:20 -07:00 |
|
Glenn Jocher
|
bafbc65ee3
|
AutoAnchor bug fix
|
2020-06-16 14:26:02 -07: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 |
|
Glenn Jocher
|
57a0ae3350
|
AutoAnchor implementation
|
2020-06-16 13:12:22 -07:00 |
|
Glenn Jocher
|
ec81c7b5f2
|
check_anchors() bug fix #90
|
2020-06-16 10:14:04 -07: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
|
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
|
c5966abba8
|
glob search bug fix #77
|
2020-06-15 12:08:57 -07:00 |
|
Glenn Jocher
|
260b1729f0
|
FP16 inference update
|
2020-06-14 18:32:29 -07: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 |
|
Glenn Jocher
|
099e6f5ebd
|
--img-size stride-multiple verification
|
2020-06-12 22:10:46 -07:00 |
|
Glenn Jocher
|
3a5c5328c5
|
ONNX, BCEBlurWithLogitsLoss, plot_study updates
|
2020-06-11 14:44:09 -07:00 |