Laughing
4e2b9ecc7e
LR --resume repeat bug fix ( #565 )
4 anos atrás
AlexWang1900
a209a32019
Fix bug #541 #542 ( #545 )
* fix #541 #542
* Update train.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 anos atrás
NanoCode012
7f8471eaeb
--notest bug fix ( #518 )
* Fix missing results_file and fi when notest passed
* Update train.py
reverting previous changes and removing functionality from 'if not opt.notest or final_epoch: # Calculate mAP' loop.
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 anos atrás
Glenn Jocher
9da56b62dd
v2.0 Release ( #491 )
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 anos atrás
Glenn Jocher
5e970d45c4
Update train.py ( #462 )
4 anos atrás
Glenn Jocher
3edc38f603
update train.py gsutil bucket fix ( #463 )
4 anos atrás
Glenn Jocher
776555771f
update train.py gsutil bucket fix ( #463 )
4 anos atrás
Glenn Jocher
b569ed6d6b
pretrained model loading bug fix ( #450 )
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 anos atrás
yzchen
4102fcc9a7
[WIP] Feature/ddp fixed ( #401 )
* Squashed commit of the following:
commit d738487089
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 17:33:38 2020 +0700
Adding world_size
Reduce calls to torch.distributed. For use in create_dataloader.
commit e742dd9619
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 15:38:48 2020 +0800
Make SyncBN a choice
commit e90d400438
Merge: 5bf8beb cd90360
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Tue Jul 14 15:32:10 2020 +0800
Merge pull request #6 from NanoCode012/patch-5
Update train.py
commit cd9036017e
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 13:39:29 2020 +0700
Update train.py
Remove redundant `opt.` prefix.
commit 5bf8bebe88
Merge: c9558a9 a1c8406
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 14:09:51 2020 +0800
Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit c9558a9b51
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 13:51:34 2020 +0800
Add device allocation for loss compute
commit 4f08c692fb
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:16:27 2020 +0800
Revert drop_last
commit 1dabe33a5a
Merge: a1ce9b1 4b8450b
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:15:49 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit a1ce9b1e96
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:15:21 2020 +0800
fix lr warning
commit 4b8450b46d
Merge: b9a50ae 02c63ef
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Wed Jul 8 21:24:24 2020 +0800
Merge pull request #4 from NanoCode012/patch-4
Add drop_last for multi gpu
commit 02c63ef81c
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Wed Jul 8 10:08:30 2020 +0700
Add drop_last for multi gpu
commit b9a50aed48
Merge: ec2dc6c 121d90b
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:48:04 2020 +0800
Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit ec2dc6cc56
Merge: d0326e3 82a6182
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:34:31 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit d0326e398d
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:31:24 2020 +0800
Add SyncBN
commit 82a6182b3a
Merge: 96fa40a 050b2a5
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Tue Jul 7 19:21:01 2020 +0800
Merge pull request #1 from NanoCode012/patch-2
Convert BatchNorm to SyncBatchNorm
commit 050b2a5a79
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 12:38:14 2020 +0700
Add cleanup for process_group
commit 2aa330139f
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 12:07:40 2020 +0700
Remove apex.parallel. Use torch.nn.parallel
For future compatibility
commit 77c8e27e60
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 01:54:39 2020 +0700
Convert BatchNorm to SyncBatchNorm
commit 96fa40a3a9
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Mon Jul 6 21:53:56 2020 +0800
Fix the datset inconsistency problem
commit 16e7c269d0
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Mon Jul 6 11:34:03 2020 +0800
Add loss multiplication to preserver the single-process performance
commit e838055630
Merge: 625bb49 3bdea3f
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Fri Jul 3 20:56:30 2020 +0800
Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit 625bb49f4e
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 2 22:45:15 2020 +0800
DDP established
* Squashed commit of the following:
commit 94147314e5
Merge: 65157e2 37acbdc
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 16 14:00:17 2020 +0800
Merge branch 'master' of https://github.com/ultralytics/yolov4 into feature/DDP_fixed
commit 37acbdc0b6
Author: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed Jul 15 20:03:41 2020 -0700
update test.py --save-txt
commit b8c2da4a0d
Author: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed Jul 15 20:00:48 2020 -0700
update test.py --save-txt
commit 65157e2fc9
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Wed Jul 15 16:44:13 2020 +0800
Revert the README.md removal
commit 1c802bfa50
Merge: cd55b44 0f3b8bb
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Wed Jul 15 16:43:38 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit cd55b445c4
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Wed Jul 15 16:42:33 2020 +0800
fix the DDP performance deterioration bug.
commit 0f3b8bb1fa
Author: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed Jul 15 00:28:53 2020 -0700
Delete README.md
commit f5921ba1e3
Merge: 85ab2f3 bd3fdbb
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Wed Jul 15 11:20:17 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit bd3fdbbf1b
Author: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue Jul 14 18:38:20 2020 -0700
Update README.md
commit c1a97a7767
Merge: 2bf86b8 f796708
Author: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue Jul 14 18:36:53 2020 -0700
Merge branch 'master' into feature/DDP_fixed
commit 2bf86b892fa2fd712f6530903a0d9b8533d7447a
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 22:18:15 2020 +0700
Fixed world_size not found when called from test
commit 85ab2f38cdda28b61ad15a3a5a14c3aafb620dc8
Merge: 5a19011 c8357ad
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 22:19:58 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit 5a19011949
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 22:19:15 2020 +0800
Add assertion for <=2 gpus DDP
commit c8357ad5b1
Merge: e742dd9 787582f
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Tue Jul 14 22:10:02 2020 +0800
Merge pull request #8 from MagicFrogSJTU/NanoCode012-patch-1
Modify number of dataloaders' workers
commit 787582f972
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 20:38:58 2020 +0700
Fixed issue with single gpu not having world_size
commit 6364892528
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 19:16:15 2020 +0700
Add assert message for clarification
Clarify why assertion was thrown to users
commit 69364d6050
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 17:36:48 2020 +0700
Changed number of workers check
commit d738487089
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 17:33:38 2020 +0700
Adding world_size
Reduce calls to torch.distributed. For use in create_dataloader.
commit e742dd9619
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 15:38:48 2020 +0800
Make SyncBN a choice
commit e90d400438
Merge: 5bf8beb cd90360
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Tue Jul 14 15:32:10 2020 +0800
Merge pull request #6 from NanoCode012/patch-5
Update train.py
commit cd9036017e
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 14 13:39:29 2020 +0700
Update train.py
Remove redundant `opt.` prefix.
commit 5bf8bebe88
Merge: c9558a9 a1c8406
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 14:09:51 2020 +0800
Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit c9558a9b51
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 14 13:51:34 2020 +0800
Add device allocation for loss compute
commit 4f08c692fb
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:16:27 2020 +0800
Revert drop_last
commit 1dabe33a5a
Merge: a1ce9b1 4b8450b
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:15:49 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit a1ce9b1e96
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 9 11:15:21 2020 +0800
fix lr warning
commit 4b8450b46d
Merge: b9a50ae 02c63ef
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Wed Jul 8 21:24:24 2020 +0800
Merge pull request #4 from NanoCode012/patch-4
Add drop_last for multi gpu
commit 02c63ef81c
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Wed Jul 8 10:08:30 2020 +0700
Add drop_last for multi gpu
commit b9a50aed48
Merge: ec2dc6c 121d90b
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:48:04 2020 +0800
Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit ec2dc6cc56
Merge: d0326e3 82a6182
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:34:31 2020 +0800
Merge branch 'feature/DDP_fixed' of https://github.com/MagicFrogSJTU/yolov5 into feature/DDP_fixed
commit d0326e398d
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Tue Jul 7 19:31:24 2020 +0800
Add SyncBN
commit 82a6182b3a
Merge: 96fa40a 050b2a5
Author: yzchen <Chenyzsjtu@gmail.com>
Date: Tue Jul 7 19:21:01 2020 +0800
Merge pull request #1 from NanoCode012/patch-2
Convert BatchNorm to SyncBatchNorm
commit 050b2a5a79
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 12:38:14 2020 +0700
Add cleanup for process_group
commit 2aa330139f
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 12:07:40 2020 +0700
Remove apex.parallel. Use torch.nn.parallel
For future compatibility
commit 77c8e27e60
Author: NanoCode012 <kevinvong@rocketmail.com>
Date: Tue Jul 7 01:54:39 2020 +0700
Convert BatchNorm to SyncBatchNorm
commit 96fa40a3a9
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Mon Jul 6 21:53:56 2020 +0800
Fix the datset inconsistency problem
commit 16e7c269d0
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Mon Jul 6 11:34:03 2020 +0800
Add loss multiplication to preserver the single-process performance
commit e838055630
Merge: 625bb49 3bdea3f
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Fri Jul 3 20:56:30 2020 +0800
Merge branch 'master' of https://github.com/ultralytics/yolov5 into feature/DDP_fixed
commit 625bb49f4e
Author: yizhi.chen <chenyzsjtu@outlook.com>
Date: Thu Jul 2 22:45:15 2020 +0800
DDP established
* Fixed destroy_process_group in DP mode
* Update torch_utils.py
* Update utils.py
Revert build_targets() to current master.
* Update datasets.py
* Fixed world_size attribute not found
Co-authored-by: NanoCode012 <kevinvong@rocketmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 anos atrás
Glenn Jocher
07493a715c
update train.py class count assertion #424
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
4 anos atrás
Glenn Jocher
912c06a67c
update train.py class count assertion #424
4 anos atrás
Glenn Jocher
65857adf52
update train.py ckpt loading
4 anos atrás
Jirka
e8ea772384
revert test module to confuse users...
4 anos atrás
Jirka
ef6f5b33a8
rename test
4 anos atrás
Jirka
9a880a4554
rename eval
4 anos atrás
Jirka
df270ea9d7
rename eval
4 anos atrás
Glenn Jocher
03489aaafb
comment tb_writer.add_hparams(hyp, {})
4 anos atrás
Glenn Jocher
120d40c06a
Update train.py
This updates the PR to a one-liner to minimize additions. Perhaps we can include opt in the future but let's start with this for now.
4 anos atrás
Janko Ondras
38acc5f3c5
Fix img_size naming in hyperparameters logging
4 anos atrás
Glenn Jocher
a1c8406af3
EMA and non_blocking=True
4 anos atrás
Glenn Jocher
140d84cca1
comment updates
4 anos atrás
Janko Ondras
e558963244
Log hyperparameters in tensorboard
Log both hyperparameters and command line options in tensorboard.
4 anos atrás
Glenn Jocher
01a73ec08e
multi-gpu ckpt filesize bug fix #253
4 anos atrás
Glenn Jocher
a586751904
multi-gpu ckpt filesize bug fix #253
4 anos atrás
Glenn Jocher
5de4e25d68
update tensorboard metric
4 anos atrás
Glenn Jocher
98fc483abc
train.py results.txt to bucket bug fix
4 anos atrás
Glenn Jocher
1c13e67b33
evolution bug fix #346
4 anos atrás
Laughing
51a161d9e1
Update train.py
4 anos atrás
Glenn Jocher
e16e9e43e1
new nc=len(names) check
4 anos atrás
Glenn Jocher
cb527d3af9
new nc=len(names) check
4 anos atrás
Glenn Jocher
603ea0bfdc
update log_dir to runs/exp #107
4 anos atrás
Glenn Jocher
72d5b58b9a
disable LR plot to suppress warning message
4 anos atrás
Glenn Jocher
24c5a941f0
--resume EMA fix #292
4 anos atrás
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
4 anos atrás
Glenn Jocher
bf6f41567a
hyperparameter printout update
4 anos atrás
Glenn Jocher
6b134d93c5
Update train.py
4 anos atrás
Glenn Jocher
16f6834486
update train.py and experimental.py
4 anos atrás
Alex Stoken
52bac22f09
Add in --resume functionality with option to specify path or to get most recent run
4 anos atrás
Laughing
956511dafd
fix LR bug
4 anos atrás
Glenn Jocher
bb3c346916
model.yaml nc inherited from dataset.yaml
4 anos atrás
Glenn Jocher
df224a0d8f
EMA bug fix #279
4 anos atrás
Glenn Jocher
3bdea3f697
strip_optimizer() bug fix #253
4 anos atrás
Glenn Jocher
597ed4ce63
Update train.py
4 anos atrás
Glenn Jocher
9705c15da1
Update train.py
4 anos atrás
ChristopherSTAN
a8d17ce712
Update train.py
4 anos atrás
Glenn Jocher
13f69777a6
typo fix
4 anos atrás
yxNONG
1aa2b67933
Update train.py
4 anos atrás
Glenn Jocher
86784cfdbf
--resume bug fix #252
4 anos atrás
Glenn Jocher
ad4c22cbfe
--resume bug fix #187
4 anos atrás
Glenn Jocher
3b16c865f0
assert --epochs
4 anos atrás