xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
yolov10/docs/reference/yolo/utils/torch_utils.md
Glenn Jocher bd0f7ecf6f
Docstrings arguments cleanup (#3229) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-06-17 16:17:07 +02:00

2.5 KiB
Raw Blame History

description keywords
Optimize your PyTorch models with Ultralytics YOLO's torch_utils functions such as ModelEMA, select_device, and is_parallel. Ultralytics YOLO, Torch, Utils, Pytorch, Object Detection

ModelEMA

::: ultralytics.yolo.utils.torch_utils.ModelEMA



EarlyStopping

::: ultralytics.yolo.utils.torch_utils.EarlyStopping



torch_distributed_zero_first

::: ultralytics.yolo.utils.torch_utils.torch_distributed_zero_first



smart_inference_mode

::: ultralytics.yolo.utils.torch_utils.smart_inference_mode



select_device

::: ultralytics.yolo.utils.torch_utils.select_device



time_sync

::: ultralytics.yolo.utils.torch_utils.time_sync



fuse_conv_and_bn

::: ultralytics.yolo.utils.torch_utils.fuse_conv_and_bn



fuse_deconv_and_bn

::: ultralytics.yolo.utils.torch_utils.fuse_deconv_and_bn



model_info

::: ultralytics.yolo.utils.torch_utils.model_info



get_num_params

::: ultralytics.yolo.utils.torch_utils.get_num_params



get_num_gradients

::: ultralytics.yolo.utils.torch_utils.get_num_gradients



model_info_for_loggers

::: ultralytics.yolo.utils.torch_utils.model_info_for_loggers



get_flops

::: ultralytics.yolo.utils.torch_utils.get_flops



get_flops_with_torch_profiler

::: ultralytics.yolo.utils.torch_utils.get_flops_with_torch_profiler



initialize_weights

::: ultralytics.yolo.utils.torch_utils.initialize_weights



scale_img

::: ultralytics.yolo.utils.torch_utils.scale_img



make_divisible

::: ultralytics.yolo.utils.torch_utils.make_divisible



copy_attr

::: ultralytics.yolo.utils.torch_utils.copy_attr



get_latest_opset

::: ultralytics.yolo.utils.torch_utils.get_latest_opset



intersect_dicts

::: ultralytics.yolo.utils.torch_utils.intersect_dicts



is_parallel

::: ultralytics.yolo.utils.torch_utils.is_parallel



de_parallel

::: ultralytics.yolo.utils.torch_utils.de_parallel



one_cycle

::: ultralytics.yolo.utils.torch_utils.one_cycle



init_seeds

::: ultralytics.yolo.utils.torch_utils.init_seeds



strip_optimizer

::: ultralytics.yolo.utils.torch_utils.strip_optimizer



profile

::: ultralytics.yolo.utils.torch_utils.profile