Elgato_dark/ultralytics
1
0
Fork 0
mirror of https://github.com/ultralytics/ultralytics synced 2026-04-21 22:17:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8
ultralytics/docs/en/reference/utils/torch_utils.md
Glenn Jocher 3637f509dd
ultralytics 8.3.231 New Zensical docs refresh (#22761) 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
2025-11-23 19:53:50 +01:00

2.9 KiB
Raw Blame History

description keywords
Explore valuable torch utilities from Ultralytics for optimized model performance, including device selection, model fusion, and inference optimization. Ultralytics, torch utils, model optimization, device selection, inference optimization, model fusion, CPU info, PyTorch tools

Reference for ultralytics/utils/torch_utils.py

!!! success "Improvements"

This page is sourced from [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/torch_utils.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/torch_utils.py). Have an improvement or example to add? Open a [Pull Request](https://docs.ultralytics.com/help/contributing/) — thank you! 🙏

::: ultralytics.utils.torch_utils.ModelEMA




::: ultralytics.utils.torch_utils.EarlyStopping




::: ultralytics.utils.torch_utils.torch_distributed_zero_first




::: ultralytics.utils.torch_utils.smart_inference_mode




::: ultralytics.utils.torch_utils.autocast




::: ultralytics.utils.torch_utils.get_cpu_info




::: ultralytics.utils.torch_utils.get_gpu_info




::: ultralytics.utils.torch_utils.select_device




::: ultralytics.utils.torch_utils.time_sync




::: ultralytics.utils.torch_utils.fuse_conv_and_bn




::: ultralytics.utils.torch_utils.fuse_deconv_and_bn




::: ultralytics.utils.torch_utils.model_info




::: ultralytics.utils.torch_utils.get_num_params




::: ultralytics.utils.torch_utils.get_num_gradients




::: ultralytics.utils.torch_utils.model_info_for_loggers




::: ultralytics.utils.torch_utils.get_flops




::: ultralytics.utils.torch_utils.get_flops_with_torch_profiler




::: ultralytics.utils.torch_utils.initialize_weights




::: ultralytics.utils.torch_utils.scale_img




::: ultralytics.utils.torch_utils.copy_attr




::: ultralytics.utils.torch_utils.intersect_dicts




::: ultralytics.utils.torch_utils.is_parallel




::: ultralytics.utils.torch_utils.unwrap_model




::: ultralytics.utils.torch_utils.one_cycle




::: ultralytics.utils.torch_utils.init_seeds




::: ultralytics.utils.torch_utils.unset_deterministic




::: ultralytics.utils.torch_utils.strip_optimizer




::: ultralytics.utils.torch_utils.convert_optimizer_state_dict_to_fp16




::: ultralytics.utils.torch_utils.cuda_memory_usage




::: ultralytics.utils.torch_utils.profile_ops




::: ultralytics.utils.torch_utils.attempt_compile