a Df @sTdZddlmZddlZddlZddlmZmZddl m Z ddd Z dd d Z dS)z Auto-batch utils )deepcopyN)LOGGERcolorstr)profileTcCsDtjj|"tt||WdS1s60YdS)N)torchcudaampautocast autobatchrtrain)modelimgszr rE/data/NAS7/SABIOD/METHODE/ermites/yolo-dyni/yolov5/utils/autobatch.pycheck_train_batch_sizesr皙?cs^td}t|dt|j}|jdkrLt|d||Stjj j rnt|d||Sd}t | }tj |}|j|} tj ||} tj ||} | | | } t||d|jd| d d | d d | d d | d d gd} z&fdd| D}t||d|d}Wn8tyb}zt||WYd}~n d}~00dd|D}tj| dt||dd}t| ||d|d}d|vr|d}|| |kr| t|dd}|dks|dkr|}t|dt||| | | }t|d|d|d| |d d| d d|ddd |S)!Nz AutoBatch: z)Computing optimal batch size for --imgsz cpuz0CUDA not detected, using default CPU batch-size uP ⚠️ Requires torch.backends.cudnn.benchmark=False, using default batch-size i@z (z) z.2fz G total, z G reserved, z G allocated, zG free)rcsg|]}t|dqS))rempty).0brrr 5zautobatch..r)ndevicecSsg|]}|r|dqS)rr)rxrrrr;rr)degriuVWARNING ⚠️ CUDA anomaly detected, recommend restart environment and retry command.zUsing batch-size z for  zG/zG (dz.0fu%) ✅)rrinfonext parametersr!typerbackendscudnn benchmarkstrupperrget_device_properties total_memorymemory_reservedmemory_allocatednamer Exceptionwarningnppolyfitlenintindexmaxpolyval)r rfraction batch_sizeprefixr!gbd propertiestraf batch_sizesimgresultseyprirrrr sH       <(  <r )rT)rrr) __doc__copyrnumpyr6r utils.generalrrutils.torch_utilsrrr rrrrs