Glenn Jocher
df224a0d8f
EMA bug fix #279
преди 4 години
Glenn Jocher
3bdea3f697
strip_optimizer() bug fix #253
преди 4 години
Glenn Jocher
597ed4ce63
Update train.py
преди 4 години
Glenn Jocher
9705c15da1
Update train.py
преди 4 години
ChristopherSTAN
a8d17ce712
Update train.py
преди 4 години
Glenn Jocher
13f69777a6
typo fix
преди 4 години
yxNONG
1aa2b67933
Update train.py
преди 4 години
Glenn Jocher
86784cfdbf
--resume bug fix #252
преди 4 години
Glenn Jocher
ad4c22cbfe
--resume bug fix #187
преди 4 години
Glenn Jocher
3b16c865f0
assert --epochs
преди 4 години
yxNONG
cdb9bde181
Unify the check point of single and multi GPU
save the model.hyp etc to checkpoint when use multi GPU training
преди 4 години
Glenn Jocher
b203c9b7ff
update train.py incompatible model message fix #222
преди 4 години
Glenn Jocher
37e13f8846
update mosaic border
преди 4 години
Glenn Jocher
22fb2b0c25
refactor dataloader
преди 4 години
Glenn Jocher
256a3e89d2
small dataset bug fix #140
преди 4 години
Glenn Jocher
b50fdf16af
model.names multi-GPU bug fix #94
преди 4 години
Glenn Jocher
b8557f87e3
add stride to datasets.py
преди 4 години
Glenn Jocher
6c1b87a42e
update google_utils import
преди 4 години
Glenn Jocher
1f1917ef56
remove fast, add merge
преди 4 години
Lornatang
899f1d4bde
Fix DDP bug in single process multiple device use cases
преди 4 години
Glenn Jocher
cce95e744d
backbone as FP16, save default to FP32
преди 4 години
Glenn Jocher
d9b64c27c2
save ckpt in FP16 #119
преди 4 години
Glenn Jocher
9fdb0fbacf
AutoAnchor bug fix # 117
преди 4 години
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 години
Glenn Jocher
8db51c7002
tb_writer bug fix
преди 4 години
Glenn Jocher
afe1df385b
dist.destroy_process_group() bug fix
преди 4 години
Glenn Jocher
5a50491fa1
check_anchors bug fix
преди 4 години
Glenn Jocher
05b8ee5ca4
check_anchors() bug fix #102
преди 4 години
Glenn Jocher
ec81c7b5f2
check_anchors() bug fix #90
преди 4 години
Glenn Jocher
8b26e89006
AutoAnchor bug fix #72
преди 4 години
Glenn Jocher
bdd9fee841
update fast mode
преди 4 години
Glenn Jocher
1c0b6236e3
update fast mode
преди 4 години
Glenn Jocher
915b1481fc
default check_git_status() to True
преди 4 години
Glenn Jocher
14523bb030
FP16 to FP32 ckpt load
преди 4 години
Glenn Jocher
c5966abba8
glob search bug fix #77
преди 4 години
Glenn Jocher
31f3310029
assert best possible recall > 0.9 before training
преди 4 години
Lauritzen Kasper Primdal
c3d4d321d3
Ensures weights/ dir exists
Allows train.py to be run outside of yolov5/ directory.
преди 4 години
Glenn Jocher
099e6f5ebd
--img-size stride-multiple verification
преди 4 години
Glenn Jocher
22d6088205
speed-reproducibility fix #17
преди 4 години
Glenn Jocher
55ca5c74d2
multi-scale fix #16
преди 4 години
Glenn Jocher
7c2832cd49
assert equal model and dataset classes
преди 4 години
Glenn Jocher
11121e39ed
updates
преди 4 години
Glenn Jocher
eb97b2e413
NMS fast mode
преди 4 години
Glenn Jocher
ce36905358
updates
преди 4 години
Glenn Jocher
1e84a23f38
initial commit
преди 4 години