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