HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3
ultralytics-ascend/docs/en/reference/utils/torch_utils.md
Laughing 0fa1d7d5a6
ultralytics 8.3.29 Sony IMX500 export (#14878) 
Signed-off-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>
Co-authored-by: Francesco Mattioli <Francesco.mttl@gmail.com>
Co-authored-by: Lakshantha Dissanayake <lakshantha@ultralytics.com>
Co-authored-by: Lakshantha Dissanayake <lakshanthad@yahoo.com>
Co-authored-by: Chizkiyahu Raful <37312901+Chizkiyahu@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Muhammad Rizwan Munawar <muhammadrizwanmunawar123@gmail.com>
Co-authored-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
2024-11-11 14:20:21 +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

!!! note

This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/torch_utils.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/torch_utils.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/utils/torch_utils.py) 🛠️. Thank you 🙏!

::: ultralytics.utils.torch_utils.ModelEMA




::: ultralytics.utils.torch_utils.EarlyStopping




::: ultralytics.utils.torch_utils.FXModel




::: 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.get_latest_opset




::: ultralytics.utils.torch_utils.intersect_dicts




::: ultralytics.utils.torch_utils.is_parallel




::: ultralytics.utils.torch_utils.de_parallel




::: ultralytics.utils.torch_utils.one_cycle




::: ultralytics.utils.torch_utils.init_seeds




::: ultralytics.utils.torch_utils.strip_optimizer




::: ultralytics.utils.torch_utils.convert_optimizer_state_dict_to_fp16




::: ultralytics.utils.torch_utils.profile