Chiebot-Mirror
/
ultralytics
mirror of https://github.com/ultralytics/ultralytics.git
9
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
705 Commits
95 Branches
408 Tags
3.8 GiB
 
 
 
Tag: Branch: Tree: 14c05f0dd1
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '14c05f0dd1'
${ noResults }
ultralytics/docs/en/reference/utils/torch_utils.md

2.6 KiB
Raw Blame History Escape

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

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). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/utils/torch_utils.py) 🛠️. 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