HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3
ultralytics-ascend/docs/reference/utils/torch_utils.md

2.6 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

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