Commit graph

56 commits

Author SHA1 Message Date
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
Glenn Jocher
fbed8499da
ultralytics 8.1.15 add Python 3.12 compatibility (#8210)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Johnny <johnnynuca14@gmail.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-02-17 20:49:34 +01:00
Glenn Jocher
863c788f67
ultralytics 8.1.13 fix yolo train time= bug (#8179)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-02-13 20:07:54 +01:00
Glenn Jocher
70d4a3752e
ultralytics 8.1.9 replace .size(0) with .shape[0] (#7957)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-02-01 23:33:02 +01:00
Glenn Jocher
ee88882874
Add HUB-SDK Docs reference section (#7781)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
2024-01-27 00:21:31 +01:00
ExtReMLapin
c267bd3a0c
Fixed dataloader CPU bottleneck for small batch sizes (#7659)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-01-19 16:49:54 +01:00
Glenn Jocher
fe27db2f6e
ultralytics 8.0.239 Ultralytics Actions and hub-sdk adoption (#7431)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
2024-01-10 03:16:08 +01:00
Adamcode
eeea5de623
Fix Multi-GPU resume bug (#7328)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-01-05 16:49:17 +01:00
Glenn Jocher
072291bc78
ultralytics 8.0.235 YOLOv8 OBB train, val, predict and export (#4499)
Co-authored-by: Yash Khurana <ykhurana6@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Swamita Gupta <swamita2001@gmail.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Laughing-q <1182102784@qq.com>
2024-01-05 03:00:26 +01:00
Glenn Jocher
6cbe736bfd
ultralytics 8.0.228 add training time argument (#7054)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-12-20 00:01:20 +01:00
Glenn Jocher
f2f5ed2c5e
ultralytics 8.0.206 engine Trainer updates (#6111)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: jamjamjon <51357717+jamjamjon@users.noreply.github.com>
2023-11-04 02:57:35 +01:00
Glenn Jocher
5b3c4cfc0e
ultralytics 8.0.198 MLflow fix, tests and Docs page (#5357)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-10-13 20:41:05 +02:00
Glenn Jocher
12e3eef844
ultralytics 8.0.197 save P, R, F1 curves to metrics (#5354)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: erminkev1 <83356055+erminkev1@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Andy <39454881+yermandy@users.noreply.github.com>
2023-10-13 02:49:31 +02:00
Glenn Jocher
7517667a33
Add docformatter to pre-commit (#5279)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
2023-10-09 02:25:22 +02:00
Glenn Jocher
daf16ecf31
Update MANIFEST.in (#4894)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-09-15 01:32:10 +02:00
Glenn Jocher
16ce193d6e
ultralytics 8.0.174 Tuner plots and improvements (#4799)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-09-10 03:27:23 +02:00
Glenn Jocher
dfe6dfb1d2
ultralytics 8.0.173 fix missing val plots and new Conda Guide (#4783) 2023-09-07 19:24:54 +02:00
Glenn Jocher
187b504d68
ultralytics 8.0.169TQDM, INTERP_LINEAR and RTDETR load_image() updates (#4704)
Co-authored-by: Rustem Galiullin <rustemgal@gmail.com>
Co-authored-by: Rustem Galiullin <rustem.galiullin@bayanat.ai>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-09-02 20:01:57 +02:00
Glenn Jocher
a4fabfdacf
Set workers=0 for MPS Train and Val modes (#4697) 2023-09-02 04:25:03 +02:00
Glenn Jocher
d2cf7acce0
ultralytics 8.0.167 Tuner updates and HUB Pose and Classify fixes (#4656)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-31 01:38:42 +02:00
Glenn Jocher
7e99804263
Cleanup argument handling in Model class (#4614)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Sebastian Stapf <42514241+Wiqzard@users.noreply.github.com>
2023-08-28 23:12:32 +02:00
Laughing
53b4f8c713
Fix DDP when device is a list (#4600)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-08-28 13:04:30 +02:00
Glenn Jocher
23b4f697c9
Add new get_save_dir() function (#4602) 2023-08-28 10:43:41 +02:00
Laughing
47ab96dab6
Fix pytest world_size environment bug (#4590)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-27 15:14:29 +02:00
Glenn Jocher
2bcee56e70
ultralytics 8.0.162 Multi-GPU DDP fix (#4544)
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: andresinsitu <andres.rodriguez@ingenieriainsitu.com>
2023-08-24 13:13:49 +02:00
Glenn Jocher
3c40e7a9fc
Fix TensorBoard graph UserWarning catch (#4513)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-23 12:52:24 +02:00
Glenn Jocher
9a0555eca4
ultralytics 8.0.155 allow imgsz and batch resume changes (#4366)
Co-authored-by: Mostafa Nemati <58460889+monemati@users.noreply.github.com>
Co-authored-by: Eduard Voiculescu <eduardvoiculescu95@gmail.com>
2023-08-15 22:02:23 +02:00