Commit graph

215 commits

Author SHA1 Message Date
Glenn Jocher
972f615110
New ONNX, TorchScript, CoreML export test matrices (#11652)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-05-05 16:57:52 +02:00
Burhan
2583f842b8
ultralytics 8.2.9 OpenVINO INT8 fixes and tests (#10423)
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-05-05 15:11:17 +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
Lakshantha Dissanayake
6268ac8e1e
ultralytics 8.2.7 add Raspberry Pi Benchmarks CI (#10280)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-05-03 03:25:51 +02:00
Lakshantha Dissanayake
bbfca94467
Fix TFLite export for NVIDIA Jetson (#10404)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-05-01 15:56:30 +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
03badaff57
Improve PNNX assets check robustness (#10089) 2024-04-16 16:19:31 -07:00
Mo Li
3bbbdcfef8
Fix overrides training cfg bug (#10002)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Ultralytics AI Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
Co-authored-by: Laughing-q <1185102784@qq.com>
2024-04-14 07:24:25 -07:00
Glenn Jocher
42416bc608
ultralytics 8.1.47 PNNX release 20240226 (#9954)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-04-11 18:04:54 +02:00
Burhan
4ffd6ee6d7
ultralytics 8.1.46 add TensorRT 10 support (#9516)
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: 九是否随意的称呼 <1069679911@qq.com>
2024-04-10 18:07:53 +02:00
Glenn Jocher
ea03db9984
Pin PNNX to avoid PNNX==20240410 bug (#9953)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-04-10 17:46:41 +02:00
Glenn Jocher
bc636d08b7
Improve PNNX download robustness (#9884) 2024-04-08 22:55:18 +02:00
Glenn Jocher
9b7a0c82fd
Fix missing Tuner ckpt_file (#9847) 2024-04-07 17:05:47 +02: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
2cee8893d9
Define export formats model.names attribute (#9474) 2024-04-01 18:53:44 +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
b530a3004e
Define fallback model.names from Predictor (#9439) 2024-03-31 06:33:43 +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
Jiacong Fang
03d0ffd9f5
Improved float workspace arg for TRT exports (#9407)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-29 16:22:32 +01:00
Glenn Jocher
4a7ccba0af
ultralytics 8.1.37 fix empty sys.argv bug (#9390)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Joshua Harrison <108553645+TurbineJoshua@users.noreply.github.com>
Co-authored-by: Joshua Harrison <joshua@fido.investments>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2024-03-29 03:20:39 +01: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
sifan-intel
e84e11724b
Specify YOLODataset task arg during INT8 calibration (#9309)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-27 21:00:57 +01:00
Glenn Jocher
5be2ffbd13
ultralytics 8.1.34 Inference API robust imgsz checks (#9274)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-25 02:00:38 +01:00
Kalen Michael
ec1d110689
ultralytics 8.1.33 fix HUB model checks (#9153)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-24 06:14:15 +01:00
Laughing
33607ca278
Add bgr hyperparameter (#9139)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-23 23:17:00 +01:00
Glenn Jocher
3c179f87cb
ultralytics 8.1.31 NCNN and CLIP updates (#9235)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-22 23:51:50 +01:00
Kalen Michael
8617fcf32d
ultralytics 8.1.30 add advanced HUB train arguments (#9110)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-20 15:53:00 -07:00
UltralyticsAssistant
bc34ae560b Auto-format by https://ultralytics.com/actions 2024-03-15 00:12:44 +00:00
Lakshantha Dissanayake
368abb4cc8
Optimize TFJS export on ARM64 (#8946)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-15 01:12:22 +01:00
Glenn Jocher
d24ead447e
More robust PNNX downloads (#8866) 2024-03-11 20:13:49 +01:00
Glenn Jocher
617ab21523
Faster batch same_shapes (#8851) 2024-03-11 19:53:35 +01:00
Laughing
2ea6b2b889
ultralytics 8.1.27 batched tracking fixes (#8842)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-11 19:29:41 +01:00
Glenn Jocher
7638c5ce4d
Integrate OpenVINO CUMULATIVE_THROUGHPUT mode batched inference (#8834)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-11 18:50:29 +01:00
Burhan
ff8ae0d2e2
Batch inference fixes (#8854) 2024-03-11 17:14:03 +01:00
Glenn Jocher
7451ca1f54
ultralytics 8.1.26 LoadImagesAndVideos batched inference (#8817)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-10 19:15:44 +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
Finlay Morrison
10bb39d828
Modify tuner best.pt logic to train first (#8792) 2024-03-08 15:57:12 +01:00
Johnny
b53a3989ab
OpenVino 2024.0.0 (#8726)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-06 21:55:23 +01:00
Glenn Jocher
90943946a7
ultralytics 8.1.25 OpenVINO LATENCY and THROUGHPUT modes (#8058)
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Adrian Boguszewski <adekboguszewski@gmail.com>
2024-03-06 11:38:05 +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
Adrian Boguszewski
a7cfd83c5f
ultralytics 8.1.24 new OpenVINO 2023.3 export updates (#8417)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2024-03-05 15:49:02 +01:00