Commit graph

82 commits

Author SHA1 Message Date
Laughing
60d6d011fd
Fix automatic optimizer LR with DDP training (#18842) 2025-01-23 13:26:29 +01:00
Ivor Zhu
b5e0cee943
ultralytics 8.3.65 Rockchip RKNN Integration for Ultralytics YOLO models (#16308)
Signed-off-by: Francesco Mattioli <Francesco.mttl@gmail.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
Co-authored-by: Lakshantha Dissanayake <lakshantha@ultralytics.com>
Co-authored-by: Burhan <Burhan-Q@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
Co-authored-by: Lakshantha Dissanayake <lakshanthad@yahoo.com>
Co-authored-by: Francesco Mattioli <Francesco.mttl@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2025-01-21 02:25:54 +01:00
Mohammed Yasin
d9c374c69b
Fix automatic optimizer LR with DDP training (#18641)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
2025-01-16 22:45:49 +01:00
Paula Derrenger
deb87d8ce3
Standardize license headers in Python files (#18696)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2025-01-15 15:36:11 +01:00
Glenn Jocher
3902e740cf
Apply Ruff 0.9.0 (#18622)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2025-01-10 00:05:09 +01:00
Laughing
4453ddab93
ultralytics 8.3.31 add max_num_obj factor for AutoBatch (#17514)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-13 23:51:24 +01:00
Paula Derrenger
7373816b23
Fix PyPI downloads links (#17399)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-11-07 17:15:15 +01:00
Glenn Jocher
3a4b65c347
ultralytics 8.3.27 HUB timed training fix (#17298)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-11-01 01:42:51 +01:00
Muhammad Rizwan Munawar
c943a3b747
Case-insensitive optimizer name (#17287)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-10-31 12:52:14 +01:00
Glenn Jocher
50497218c2
ultralytics 8.3.9 add training time to results CSV (#16803)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-10-09 16:19:52 +02:00
Glenn Jocher
d90efbd297
ultralytics 8.3.6 improve collect_system_info (#16722)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-10-06 20:36:35 +02:00
Laughing
db3c0400c5
Fix overridden train_args when close_mosaic (#16627)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-10-02 13:40:47 +02:00
Glenn Jocher
6e43d1e1e5
ultralytics 8.3.0 YOLO11 Models Release (#16539)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-09-30 02:59:20 +02:00
Glenn Jocher
ba438aea5a
ultralytics 8.2.95 faster checkpoint saving (#16311)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-09-16 21:37:14 +02:00
Quet Almahdi Morris
fa6362a6f5
ultralytics 8.2.94 Apple MPS train memory display (#16272)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-09-15 19:34:31 +02:00
Skillnoob
e0cbbf409d
Fix mps.empty_cache() for macOS without MPS (#16280)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-09-15 16:59:30 +02:00
Quet Almahdi Morris
4d5afa7e0d
MPS unified memory cache empty (#16078)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-09-07 19:38:03 +02:00
Laughing
7c7f456710
Fix torch.cuda.amp.GradScaler warning (#15978)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-09-03 12:03:01 +02:00
Glenn Jocher
776ca86369
Ruff Docstring formatting (#15793)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-08-25 04:27:55 +08:00
Glenn Jocher
c1882a4327
'best.pt' inherit all-epochs results curves from 'last.pt' (#15791)
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-08-24 23:17:09 +08:00
alanZee
03e0b1033c
Improve trainer DDP device handling (#15383)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-08-13 04:53:48 +08:00
Haris Rehman
9c5d1a2451
ultralytics 8.2.71 Multinode DDP training (#14879)
Co-authored-by: Haris Rehman <haris.rehman.cowlar@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-08-01 17:31:03 +02:00
Glenn Jocher
9130399974
ultralytics 8.2.66 HUB model autodownload (#14702)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-07-26 00:31:41 +02:00
Burhan
db42d180d9
Updates save_period to include first epoch (#14700)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-07-25 23:49:04 +02:00
Laughing
1c351b5036
Fix torch.amp has no attribute GradScaler (#14647)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-07-24 10:12:19 +02:00
Glenn Jocher
0d7bf447eb
Fix torch.amp.autocast('cuda') warnings (#14633)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
2024-07-23 21:58:39 +02:00
Chia-Hsiang Tsai
470b120a1b
BaseTrainer with find_unused_parameters=True when using DistributedDataParallel() (#14323) 2024-07-10 19:59:16 +02:00
Laughing
169602442c
Fix HUB session with DDP training (#13103)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-06-23 20:00:34 +02:00
Glenn Jocher
6a234f3639
ultralytics 8.2.29 new fractional AutoBatch feature (#13446)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-06-09 00:28:11 +02:00
Glenn Jocher
2fe0946376
Fix Paddle 2.6.0 Dockerfile install bug (#13447)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-06-08 23:34:51 +02:00
Hoonjae Lee
b87ea6ab22
ultralytics 8.2.16 DDP pretrained argument fix (#11787)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
2024-05-15 19:39:26 +02:00
Frank Ruis
2f9a604387
fix double zero_grad call messing up gradient accumulation (#11217)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-05-03 23:00:18 +02:00
Glenn Jocher
1110258d37
ultralytics 8.2.2 replace COCO128 with COCO8 (#10167)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-04-18 20:47:21 -07:00
Glenn Jocher
1f4bed233a
Implement gc.collect() to free memory (#10129)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-04-17 19:37:40 -07:00
Glenn Jocher
7d891a4aa4
ultralytics 8.1.44 add IS_RASPBERRYPI and constants refactor (#9827)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-04-07 00:47:12 +02:00
Glenn Jocher
a2628657a1
ultralytics 8.1.43 40% faster ultralytics imports (#9547)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-04-05 15:29:09 +02:00
Laughing
1e547e60a0
Fix learning rate gap on resume (#9468)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: EunChan Kim <eunchan@hanyang.ac.kr>
Co-authored-by: Lakshantha Dissanayake <lakshanthad@yahoo.com>
Co-authored-by: RizwanMunawar <chr043416@gmail.com>
Co-authored-by: gs80140 <gs80140@users.noreply.github.com>
2024-04-02 11:55:11 +02:00
Glenn Jocher
959acf67db
ultralytics 8.1.41 DDP resume untrained-checkpoint fix (#9453)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
2024-04-01 19:46:04 +02:00
Glenn Jocher
ea527507fe
ultralytics 8.1.40 search in Python sets {} for speed (#9450)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-04-01 00:16:52 +02:00
Glenn Jocher
7df821e6ea
Deprecations fix TORCH_NCCL_BLOCKING_WAIT (#9448) 2024-03-31 18:36:50 +02:00
Laughing
e9187c1296
ultralytics 8.1.39 add YOLO-World training (#9268)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-31 16:30:17 +02:00
Glenn Jocher
1703025e8e
Save optimizer as FP16 for smaller checkpoints (#9435)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-03-31 15:11:42 +02:00
Glenn Jocher
7ea2007326
Favor EMA over model in train checkpoints (#9433)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-31 06:14:22 +02:00
Glenn Jocher
479afce4a8
Immediate checkpoint serialization (#9437) 2024-03-31 05:53:34 +02:00
Glenn Jocher
eed01a2cf3
Faster best.pt checkpoint saving (#9436) 2024-03-31 05:01:30 +02:00
Glenn Jocher
ed2250cf1c
ultralytics 8.1.36 improve train stop robustness epoch + 1 >= self.epochs (#9384)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-28 17:28:20 +01:00
Glenn Jocher
2bc605f32a
ultralytics 8.1.25 fix **kwargs: (dict) warnings (#8815)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-09 18:51:38 +01:00
Laughing
ddc94a6981
OBB: Fix when training on zip data (#8680)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-05 18:37:33 +01:00
Laughing
946e18f79c
ultralytics 8.1.21 Add YOLOv8-World-v2 models (#8580)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-03-03 22:56:57 +01:00
Alexander Suslov
5171d6679d
Added check of parameter type before setting required_grad=True for frozen layers (#8318)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-02-20 17:32:04 +01:00