ultralytics/docs/reference/utils/torch_utils.md

ModelEMA

---

::: ultralytics.utils.torch_utils.ModelEMA

EarlyStopping

---

::: ultralytics.utils.torch_utils.EarlyStopping

torch_distributed_zero_first

---

::: ultralytics.utils.torch_utils.torch_distributed_zero_first

smart_inference_mode

---

::: ultralytics.utils.torch_utils.smart_inference_mode

get_cpu_info

---

::: ultralytics.utils.torch_utils.get_cpu_info

select_device

---

::: ultralytics.utils.torch_utils.select_device

time_sync

---

::: ultralytics.utils.torch_utils.time_sync

fuse_conv_and_bn

---

::: ultralytics.utils.torch_utils.fuse_conv_and_bn

fuse_deconv_and_bn

---

::: ultralytics.utils.torch_utils.fuse_deconv_and_bn

model_info

---

::: ultralytics.utils.torch_utils.model_info

get_num_params

---

::: ultralytics.utils.torch_utils.get_num_params

get_num_gradients

---

::: ultralytics.utils.torch_utils.get_num_gradients

model_info_for_loggers

---

::: ultralytics.utils.torch_utils.model_info_for_loggers

get_flops

---

::: ultralytics.utils.torch_utils.get_flops

get_flops_with_torch_profiler

---

::: ultralytics.utils.torch_utils.get_flops_with_torch_profiler

initialize_weights

---

::: ultralytics.utils.torch_utils.initialize_weights

scale_img

---

::: ultralytics.utils.torch_utils.scale_img

make_divisible

---

::: ultralytics.utils.torch_utils.make_divisible

copy_attr

---

::: ultralytics.utils.torch_utils.copy_attr

get_latest_opset

---

::: ultralytics.utils.torch_utils.get_latest_opset

intersect_dicts

---

::: ultralytics.utils.torch_utils.intersect_dicts

is_parallel

---

::: ultralytics.utils.torch_utils.is_parallel

de_parallel

---

::: ultralytics.utils.torch_utils.de_parallel

one_cycle

---

::: ultralytics.utils.torch_utils.one_cycle

init_seeds

---

::: ultralytics.utils.torch_utils.init_seeds

strip_optimizer

---

::: ultralytics.utils.torch_utils.strip_optimizer

profile

---

::: ultralytics.utils.torch_utils.profile