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 7565210484
ultralytics 8.0.149 add Open Images V7 training (#4178) 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: AdiEcho <30563671+AdiEcho@users.noreply.github.com>
2023-08-07 01:19:51 +02:00

2.5 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). Help us fix any issues you see by submitting 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.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