\;j:ddlZddlmZGddZGddZGddZGd d eZGd d ZGd dZeZ dZ y)N) OrderedDictc.eZdZdZdZdZdZdZdZy)Stackz The stack in a Last-In/First-Out (LIFO) manner. New element is added at the end and an element is removed from that end. cg|_yN)itemsselfs ^G:\00. PROJECTS\API\Inventory\templateJSON\kerjaOCR\Lib\site-packages\paddle/profiler/timer.py__init__zStack.__init__s  c:|jj|yr)rappend)r items r pushz Stack.pushs $r c6|jjSr)rpopr s r rz Stack.popszz~~r c2t|jdk(SNr)lenrr s r is_emptyzStack.is_empty"s4::!##r cn|js%|jt|jdz SyN)rrrr s r peekz Stack.peek%s*}}::c$**o12 2r N) __name__ __module__ __qualname____doc__r rrrrr r rrs    $r rcHeZdZdZdZdZdZd dZdZdZ d Z d Z d Z y) Eventzt A Event is used to record the cost of every step and the cost of the total steps except skipped steps. ct|_t|_d|_d|_d|_dt ddd|_dt ddd|_dt dd|_ d|_ d|_ d|_ d|_ y)Nr inf)maxmintotal)r&r'T samples/s) TimeAveragerreader_cost_averagerbatch_cost_averager total_samples total_iters skip_iterfloatreader_records batch_records speed_recordsreader need_record speed_mode speed_unitr s r r zEvent.__init__2s$0N!#/> &'e qI%&uU|aH%&uU|< &&r cl|jj|jjyr)r+resetr,r s r r9z Event.resetDs& !!'')   &&(r c|jj||j|jk\r|j ||j yyr)r+recordr.r/_update_recordsr1)r usetimes r record_readerzEvent.record_readerHsC !!((1   t~~ -  $*=*= > .r Nc|d|_d|_|jj|||xjdz c_|j|j k\rq|j ||j|jdk(r$t||z }|xj|z c_ nd|z }|j ||jyy)Nsteps/srr)g?) r6r7r,r;r.r/r<r2r0r-r3)r r= num_samples current_speeds r record_batchzEvent.record_batchMs  'DO'DO   ''= A   t~~ -  $*<*< =+- %k 2W < ""k1" #g  0B0B C .r cz||dkDr||d<n ||dkr||d<d|jvr|dxx|z cc<yy)Nr&r'r()keys)r current_recordrecordss r r<zEvent._update_records]sO GEN *+GEN gen ,+GEN glln $ G  .  %r c6|jjSr)r+ get_averager s r reader_averagezEvent.reader_averagees((4466r c6|jjSr)r,rIr s r batch_averagezEvent.batch_averagehs''3355r c|jdk(r|jjS|jjS)Nr))r6r,get_ips_averageget_step_averager s r speed_averagezEvent.speed_averageks8 ??k )++;;= =++<<> >r c|j|jkriSd}d}d}|xj|jzc_|jdt|jz }|jdt|jz }|j dk(r&t|j |jdz }n%t|j|jdz }|jd|jd|d}|jd|jd|d}|jd|jd|d}||z dz}||||d}|S) Nrr(r)r&r')r&r'avgd)reader_summary batch_summary ips_summary reader_ratio)r.r/r1r0r2r6r-r3) r reader_avg batch_avg speed_avgrTrUrVrWsummarys r get_summaryzEvent.get_summaryqsl   t~~ -I    DNN*((1E$:J:J4KK &&w/%8H8H2II ??k )d001D4F4Fw4OOId../$2D2DW2MMI&&u-&&u-  %%e,%%e, %%e,%%e, #Y.#5 ,*&(  r r) rrrrr r9r>rCr<rJrLrPr\r r r r"r",s5 &$)? D /76? 'r r"c.eZdZdZdZdZdZdZdZy)HookzG As the base class. All types of hooks should inherit from it. cyrr r benchmarks r beginz Hook.begin r cyrr r`s r endzHook.endrcr cyrr r`s r before_readerzHook.before_readerrcr cyrr r`s r after_readerzHook.after_readerrcr cyrr r`s r after_stepzHook.after_steprcr N) rrrrrbrergrirkr r r r^r^s      r r^c@eZdZdZdZdZdZdZdZdZ dZ d Z y ) TimerHookzd A hook for recording real-time performance and the summary performance of total steps. chtj|_tj|_yr)timeit default_timer start_time start_readerr s r r zTimerHook.__init__s# ..0"002r c|jjt|jj|_t j |_y)z Create the event for timing and initialize the start time of a step. This function will be called in `Profiler.start()`. N)eventsrr"r current_eventrorprqr`s r rbzTimerHook.beginsA eg&"+"2"2"7"7"9  ..0r c6tj|_y)z Initialize the start time of the dataloader. This function will be called at the beginning of `next` method in `_DataLoaderIterMultiProcess` or `_DataLoaderIterSingleProcess`. N)rorprrr`s r rgzTimerHook.before_readers#002r ctj|jz }|j|jjr|dk(ry|jj |y)aX Record the cost of dataloader for the current step. Since the skipped steps are 10, it will update the maximum, minimum and the total time from the step 11 to the current step. This function will be called at the end of `next` method in `_DataLoaderIterMultiProcess` or `_DataLoaderIterSingleProcess`. Nr)rorprrrur5r>)r ra reader_costs r rizTimerHook.after_readersW**,t/@/@@  $ $ ,++77q --k:r c |j|jjsytj|jz }|jj ||j tj|_y)a Record the cost for the current step. It will contain the cost of the loading data if there is a dataloader. Similar to `after_reader`, it will also update the maximum, minimum and the total time from the step 11 to the current step as well as the maximum and minimum speed of the model. This function will be called in `Profiler.step()`. N)rur5rorprqrCrA)r ra batch_costs r rkzTimerHook.after_stepse  # # +''33 ))+doo= ,,Z9N9NO ..0r c|jjry|j||jj|jj |_t j|_y)a& Print the performance summary of the model and pop the current event from the events stack. Since there may be nested timing events, such as evaluation in the training process, the current event needs to be update to the event at the top of the stack. N) rtr_print_summaryrrrurorprqr`s r rez TimerHook.ends`    $ $ &  I&"+"2"2"7"7"9  ..0r c (|jj}|sytdjdd|ddk7rtdd|dzzdztd |jjztd d jd d d jd d djd d djd d |dd dk7r|j d|d|j d|d|j d|dy)Nz Perf SummaryrS=rWrzReader Ratio: z%.3f%zTime Unit: s, IPS Unit: %s|rRr&r'rTrxrzrUipsrV)rur\printcenterr7 _print_stats)r rar[s r r|zTimerHook._print_summarys))557  n##C-. > "a ' "Vw~/F%GG#M N (I,C,C,N,N O    IIbM  LL   LL   LL    # $U +q 0   mW5E-F G ,(@A %!78r c d|dz}d|dz}d|dz}td|jdd|jdd|jdd|jdd y)Nz%.5frRr&r'rr)rr)r r message_dictavg_strmax_strmin_strs r rzTimerHook._print_statssuL/0L/0L/0  KKO  NN2   NN2   NN2   r N) rrrrr rbrgrirkrer|rr r r rmrms/ 313;$1$1 96 r rmc6eZdZdZdZdZd dZdZdZdZ y) r*z> Record the cost of every step and count the average. c$|jyr)r9r s r r zTimeAverager.__init__3s  r c.d|_d|_d|_yr _total_iters _total_time_total_samplesr s r r9zTimeAverager.reset6sr Nc|xjdz c_|xj|z c_|r|xj|z c_yyrr)r r=rAs r r;zTimeAverager.record;s= Q G#    ; .  r cf|jdk(ry|jt|jz S)zA Get the average cost of loading data or a step. r)rrr0r s r rIzTimeAverager.get_averageAs1    !%(9(9":::r c~|jr|jdk(ryt|j|jz S)zL Get the average throughput when speed mode is "samples/s". r)rrr0rr s r rNzTimeAverager.get_ips_averageJs9 ""d&7&71&<T(()D,<,<<??:./27""./""))22:>??:./26"".6<6J6J6L <(3 /r r) rrrrr rrrbrgrirkrerr r r rr]s5 2%$"Mr rctSr) _benchmark_r r r raras r ) ro collectionsrrr"r^rmr*rrrar r r rse#2ll^  *{ { |,;,;^[M[M|k r