yolov10/docs/reference/utils/torch_utils.md
Laughing 620f3eb218
ultralytics 8.0.136 refactor and simplify package (#3748)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-07-16 17:47:45 +02:00

2.2 KiB

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