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/utils/torch_utils.md
Glenn Jocher 11d0488bf1
Update prediction Results docs (#4139) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-03 18:44:40 +02:00

2.4 KiB
Raw Blame History

description keywords
Explore Ultralytics-tailored torch utility features like Model EMA, early stopping, smart inference, image scaling, get_flops, and many more. Ultralytics, Torch Utils, Model EMA, Early Stopping, Smart Inference, Get CPU Info, Time Sync, Fuse Deconv and bn, Get num params, Get FLOPs, Scale img, Copy attr, Intersect dicts, De_parallel, Init seeds, Profile

Reference for ultralytics/utils/torch_utils.py

!!! note

Full source code for 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).

::: 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.get_cpu_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.make_divisible



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