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
bf6f41567a
hyperparameter printout update
2020-07-08 17:21:00 -07:00
dc5e18390a
Merge branch 'master' into advanced_logging
2020-07-08 17:01:19 -07:00
6b134d93c5
Update train.py
2020-07-08 16:58:13 -07:00
16f6834486
update train.py and experimental.py
2020-07-08 14:23:34 -07:00
52bac22f09
Add in --resume functionality with option to specify path or to get most recent run
2020-07-07 10:42:28 -05:00
f517ba81c7
Merge branch 'master' into advanced_logging
2020-07-06 16:52:11 -05:00
956511dafd
fix LR bug
2020-07-05 15:08:24 +08:00
bb3c346916
model.yaml nc inherited from dataset.yaml
2020-07-04 17:51:54 -07:00
df224a0d8f
EMA bug fix #279
2020-07-03 11:56:14 -07:00
3bdea3f697
strip_optimizer() bug fix #253
2020-07-02 21:24:26 -07:00
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
597ed4ce63
Update train.py
2020-07-02 12:00:55 -07:00
13f69777a6
typo fix
2020-07-02 09:26:03 -07:00
1aa2b67933
Update train.py
2020-07-02 13:51:52 +08:00
86784cfdbf
--resume bug fix #252
2020-06-30 21:43:53 -07:00
ad4c22cbfe
--resume bug fix #187
2020-06-30 16:16:29 -07:00
3b16c865f0
assert --epochs
2020-06-30 14:08:08 -07:00
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
b203c9b7ff
update train.py incompatible model message fix #222
2020-06-29 12:45:25 -07:00
37e13f8846
update mosaic border
2020-06-27 13:50:15 -07:00
e18e6811dc
Merge branch 'master' into advanced_logging
2020-06-27 10:13:03 -05:00
22fb2b0c25
refactor dataloader
2020-06-26 18:56:13 -07:00
256a3e89d2
small dataset bug fix #140
2020-06-25 17:52:56 -07:00
b50fdf16af
model.names multi-GPU bug fix #94
2020-06-24 22:22:13 -07:00
de191655e4
Fix yaml saving (don't sort keys), reorder --opt keys, bug fix hyp dict accessor
2020-06-24 17:21:54 -05:00
2d396bea00
Fix bug in --help from percent sign in help string
2020-06-24 16:57:12 -05:00
b8557f87e3
add stride to datasets.py
2020-06-24 13:02:27 -07:00
611aacf1bf
Turn opt into dictionary before sending it to yaml
2020-06-24 10:49:08 -05:00
bc4ef4861b
Default optimizer SGD
2020-06-24 10:07:43 -05:00
7abf202cad
Mode all optimizer settings to 'hyp.yaml', integrate proper momentum with Adam optimizer
2020-06-24 10:03:21 -05:00
7edbf6570e
Fix help message for cfg files
2020-06-24 09:45:57 -05:00
d64ad0fbf3
Remove --resume functionality and related checks/logic.
2020-06-24 09:17:27 -05:00
6c1b87a42e
update google_utils import
2020-06-22 23:00:23 -07:00
1f1917ef56
remove fast, add merge
2020-06-21 13:37:11 -07:00
e572bb0803
Add plot_results save location to log_dir
2020-06-21 09:36:28 -05:00
899f1d4bde
Fix DDP bug in single process multiple device use cases
2020-06-20 13:00:03 +08:00
cdf1eac9f7
Merge pull request #107 from Lornatang/fix-reference-bugs
...
fix refrence bug
2020-06-19 14:51:41 -07:00
cce95e744d
backbone as FP16, save default to FP32
2020-06-18 00:13:18 -07:00
d9b64c27c2
save ckpt in FP16 #119
2020-06-17 22:34:13 -07:00
9fdb0fbacf
AutoAnchor bug fix # 117
2020-06-17 19:51:15 -07:00
c8152c81a6
Syntax fixes
2020-06-17 16:32:13 -05:00
9b7386f603
Add save_dir arg to test.test, use arg as location for saving batch jpgs
2020-06-17 16:08:46 -05:00
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
3b2b330872
Move results.txt from weights/ to log_dir
2020-06-17 15:55:45 -05:00
ade023cff2
Fix hyp file read in and dict update.
...
Add example of hyp yaml
2020-06-17 10:59:20 -05:00
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.
2020-06-17 09:56:26 +08:00
8db51c7002
tb_writer bug fix
2020-06-16 16:05:28 -07:00
5f2eeba233
remove old print statements
2020-06-16 17:09:39 -05:00
afe1df385b
dist.destroy_process_group() bug fix
2020-06-16 15:08:14 -07:00
Alex Stoken
Glenn Jocher
Laughing
yxNONG
Lornatang