Laughing
|
51a161d9e1
|
Update train.py
|
2020-07-10 11:37:58 +08:00 |
Glenn Jocher
|
e16e9e43e1
|
new nc=len(names) check
|
2020-07-09 17:10:43 -07:00 |
|
Glenn Jocher
|
cb527d3af9
|
new nc=len(names) check
|
2020-07-09 17:03:12 -07:00 |
|
Glenn Jocher
|
603ea0bfdc
|
update log_dir to runs/exp #107
|
2020-07-09 15:58:07 -07:00 |
|
Glenn Jocher
|
72d5b58b9a
|
disable LR plot to suppress warning message
|
2020-07-09 15:16:57 -07:00 |
|
Glenn Jocher
|
24c5a941f0
|
--resume EMA fix #292
|
2020-07-09 15:09:06 -07:00 |
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
|
2020-07-09 16:18:55 -05: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
|
6b134d93c5
|
Update train.py
|
2020-07-08 16:58:13 -07:00 |
|
Glenn Jocher
|
16f6834486
|
update train.py and experimental.py
|
2020-07-08 14:23:34 -07:00 |
|
Alex Stoken
|
52bac22f09
|
Add in --resume functionality with option to specify path or to get most recent run
|
2020-07-07 10:42:28 -05:00 |
|
Alex Stoken
|
f517ba81c7
|
Merge branch 'master' into advanced_logging
|
2020-07-06 16:52:11 -05:00 |
|
Laughing
|
956511dafd
|
fix LR bug
|
2020-07-05 15:08:24 +08:00 |
|
Glenn Jocher
|
bb3c346916
|
model.yaml nc inherited from dataset.yaml
|
2020-07-04 17:51:54 -07:00 |
|
Glenn Jocher
|
df224a0d8f
|
EMA bug fix #279
|
2020-07-03 11:56:14 -07:00 |
|
Glenn Jocher
|
3bdea3f697
|
strip_optimizer() bug fix #253
|
2020-07-02 21:24:26 -07:00 |
|
Glenn Jocher
|
e02a189a3a
|
Merge pull request #245 from yxNONG/patch-2
Unify the check point of single and multi GPU
|
2020-07-02 12:05:46 -07:00 |
|
Glenn Jocher
|
597ed4ce63
|
Update train.py
|
2020-07-02 12:00:55 -07:00 |
|
Glenn Jocher
|
13f69777a6
|
typo fix
|
2020-07-02 09:26:03 -07:00 |
yxNONG
|
1aa2b67933
|
Update train.py
|
2020-07-02 13:51:52 +08:00 |
|
Glenn Jocher
|
86784cfdbf
|
--resume bug fix #252
|
2020-06-30 21:43:53 -07:00 |
|
Glenn Jocher
|
ad4c22cbfe
|
--resume bug fix #187
|
2020-06-30 16:16:29 -07:00 |
|
Glenn Jocher
|
3b16c865f0
|
assert --epochs
|
2020-06-30 14:08:08 -07:00 |
|
yxNONG
|
cdb9bde181
|
Unify the check point of single and multi GPU
save the model.hyp etc to checkpoint when use multi GPU training
|
2020-06-30 19:06:28 +08:00 |
|
Glenn Jocher
|
b203c9b7ff
|
update train.py incompatible model message fix #222
|
2020-06-29 12:45:25 -07:00 |
|
Glenn Jocher
|
37e13f8846
|
update mosaic border
|
2020-06-27 13:50:15 -07:00 |
|
Alex Stoken
|
e18e6811dc
|
Merge branch 'master' into advanced_logging
|
2020-06-27 10:13:03 -05:00 |
|
Glenn Jocher
|
22fb2b0c25
|
refactor dataloader
|
2020-06-26 18:56:13 -07:00 |
|
Glenn Jocher
|
256a3e89d2
|
small dataset bug fix #140
|
2020-06-25 17:52:56 -07:00 |
|
Glenn Jocher
|
b50fdf16af
|
model.names multi-GPU bug fix #94
|
2020-06-24 22:22:13 -07:00 |
|
Alex Stoken
|
de191655e4
|
Fix yaml saving (don't sort keys), reorder --opt keys, bug fix hyp dict accessor
|
2020-06-24 17:21:54 -05:00 |
|
Alex Stoken
|
2d396bea00
|
Fix bug in --help from percent sign in help string
|
2020-06-24 16:57:12 -05:00 |
|
Glenn Jocher
|
b8557f87e3
|
add stride to datasets.py
|
2020-06-24 13:02:27 -07:00 |
|
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 |
|
Glenn Jocher
|
6c1b87a42e
|
update google_utils import
|
2020-06-22 23:00:23 -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 |
Lornatang
|
899f1d4bde
|
Fix DDP bug in single process multiple device use cases
|
2020-06-20 13:00:03 +08: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
|
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 |