Datasets
Models
Downloads: 1
31 lines (31 with data), 4.8 kB
§
¢e® ã óR d Z ddlmZ ddlZddlZddlmZmZ ddl m
Z
d dZ d
d Z dS )z
Auto-batch utils
é )ÚdeepcopyN)ÚLOGGERÚcolorstr)Úprofileé Tc óÜ t j j |¦ « 5 t t | ¦ « ¦ « |¦ « cd d d ¦ « S # 1 swxY w Y d S )N)ÚtorchÚcudaÚampÚautocastÚ autobatchr Útrain)ÚmodelÚimgszr s ú>/home/iml/Desktop/Talha/YOLOV5_Model/yolov5/utils/autobatch.pyÚcheck_train_batch_sizer s å × Ò Ñ %Ô %ð 9ð 9Ý % ×.Ò.Ñ0Ô0°%Ñ8Ô8ð9ð 9ð 9ð 9ñ 9ô 9ð 9ð 9ð 9ð 9ð 9ð 9øøøð 9ð 9ð 9ð 9ð 9ð 9s ¥/A!Á!A%Á(A%çé?é c ó¨ t d¦ « }t j | d ¦ « t | ¦ « ¦ « j }|j dk rt j | d| ¦ « |S t j j j
rt j | d| ¦ « |S d}t |¦ « ¦ « }t j
|¦ « }|j |z } t j
|¦ « |z }
t j
|¦ « |z } | |
| z z
} t j | | d|j d| d d
|
d d | d d | d d
¦ « g d¢}
fd|
D ¦ « }t' || d|¬¦ « }n/# t( $ r"}t j | | ¦ « Y d }~nd }~ww xY wd |D ¦ « }t- j |
d t1 |¦ «
|d¬¦ « }t3 | |z |d z
|d z ¦ « }d |v r:| d ¦ « }||
| k r|
t7 |dz
d¦ « }|dk s|dk r|}t j | d¦ « t- j ||¦ « |
z | z | z }t j | d| d| d| |z d d| d d |d z d d ¦ « |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)é é é é r c ó> g | ]}t j |d ¦ « S )é )r Úempty)Ú.0Úbr s r ú
<listcomp>z autobatch.<locals>.<listcomp>5 s) ø ÐDÐDÐD°1u{1a ¨Ñ.Ô.ÐDÐDÐDó r )ÚnÚdevicec ó" g | ] }|¯|d
S )r © )r Úxs r r z autobatch.<locals>.<listcomp>; s! Ð$Ð$Ð$! !Ð$1Ð$Ð$Ð$r! r )Údegr i uV WARNING â ï¸ CUDA anomaly detected, recommend restart environment and retry command.zUsing batch-size z for ú zG/zG (éd z.0fu %) â
) r r ÚinfoÚnextÚ
parametersr# Útyper ÚbackendsÚcudnnÚ benchmarkÚstrÚupperr
Úget_device_propertiesÚ total_memoryÚmemory_reservedÚmemory_allocatedÚnamer Ú ExceptionÚwarningÚnpÚpolyfitÚlenÚintÚindexÚmaxÚpolyval)r r ÚfractionÚ
batch_sizeÚprefixr# ÚgbÚdÚ
propertiesÚtÚrÚaÚfÚ batch_sizesÚimgÚresultsÚeÚyÚpr Úis ` r r
r
sW ø õ mÑ
$Ô
$FÝ
K6ÐKÐKÀEÐKÐKÑLÔLÐLÝ
%×"Ò"Ñ$Ô$Ñ
%Ô
%Ô
,FØ
{eÒÐÝ vÐ[Ð[ÈzÐ[Ð[Ñ\Ô\Ð\ØÐÝ ~ÔÔ%ð Ý vÐ{Ð{ÐoyÐ{Ð{Ñ|Ô|Ð|ØÐð
BÝ F ×ÒÑÔAÝ×1Ò1°&Ñ9Ô9JØÔ "Ñ$AÝ
×"Ò" 6Ñ*Ô*¨RÑ/AÝ
×#Ò# FÑ+Ô+¨bÑ0AØ QU AÝ
K6Ðw1ÐwÐw
¤ÐwÐw°1ÐwÐwÐwÀ1ÐwÐwÐwÐVWÐwÐwÐwÐjkÐwÐwÐwÐwÑxÔxÐxð #Ð"Ð"Kð'ØDÐDÐDÐD¸ ÐDÑDÔDÝ#u¨°&Ð9Ñ9Ô9øÝ ð 'ð 'ð 'Ý&~ !~~Ñ&Ô&Ð&Ð&Ð&Ð&Ð&Ð&øøøøð'øøøð %Ð$wÐ$Ñ$Ô$AÝ
;w¥ A¡¤wÔ'¨°Ð2Ñ2Ô2AÝ Q\AaDÑ
A a¤DÑ (Ñ)Ô)AØ wØ MM$Ñ Ô Ø AÒ Ð ØC A¡ qMMÔ*AØ1uuDØ Ý&ÐxÐxÐxÑyÔyÐyå
1aÑ Ô 1Ñ$ qÑ(¨AÑ-HÝ
K6ÐpÐp¨AÐpÐp°AÐpÐp¸¸H¹ ÐpÐpÐpÈAÐpÐpÐpÐV^ÐadÑVdÐpÐpÐpÐpÑqÔqÐqØ Hs Å5!F Æ
GÆ!F>Æ>G)r T)r r r )
Ú__doc__Úcopyr Únumpyr: r Ú
utils.generalr r Úutils.torch_utilsr r r
r% r! r ú<module>rW s ððð ð Ð Ð Ð Ð Ð à Ð Ð Ð Ø à *Ð *Ð *Ð *Ð *Ð *Ð *Ð *Ø %Ð %Ð %Ð %Ð %Ð %ð9ð 9ð 9ð 9ð 3
ð 3
ð 3
ð 3
ð 3
ð 3
r!