fTh= S SK r S SKrS SKrS SKJr S SKJr SSKJ r SSK
J r J r S r
SS jrS r " S S
5 r " S S \5 r " S
S\ 5 rg) N)Optional )TrainerCallback)IntervalStrategy
has_lengthc p [ U 5 n U S- U S- S- U S- p2nUS:w a U SUS SUS 3$ US SUS 3$ )z$Format `t` (in seconds) to (h):mm:ssi < r :02d)int)thmss S/var/www/auris/envauris/lib/python3.13/site-packages/transformers/utils/notebook.py format_timer s[ AA4i!r'R R!A%&!VaS!C!C !CAc7!Ac71CC c & SU SU SU SU SU S3 $ )Nz
)valuetotalprefixlabelwidths r html_progress_barr ! s<
h wgeWOE7 C g r c SnUS-
nU S H nUSU S3-
nM US-
nU SS HC nUS -
nU H0 n[ U[ 5 ( a US
O
[ U5 nUS U S 3-
nM2 US
-
nME US-
nU$ )z*Put the texts in `items` in an HTML table.z%
z*
r z
| z |
z
Nz
z.6fz
| z |
z
z
)
isinstancefloatstr)items html_codeilineelts r text_to_html_tabler& , s
The main method to update the progress bar to `value`.
Args:
value (`int`):
The value to use. Must be between 0 and `total`.
force_update (`bool`, *optional*, defaults to `False`):
Whether or not to force and update of the internal state and display (by default, the bar will wait for
`value` to reach the value it predicted corresponds to a time of more than the `update_every` attribute
since the last update to avoid adding boilerplate).
comment (`str`, *optional*):
A comment to add on the left of the progress bar.
Nr r )r r2 r1 time
start_time last_time start_value elapsed_timepredicted_remainingwarmup first_callswait_for
update_barminr average_time_per_itemr+ closemaxr r6 )r7 r r: r2 current_times r updateNotebookProgressBar.update{ s
"L ?? "/3yy{ :DOdn16 6D t;? ?D 8 #{{D DM OOE "
oo
%l
T - - 1Uc$//TXTaTaBacgcmcm>n5n !# A% 99;L , >D '''-1->->%JZJZBZ-[*-1*
"
+/(zzJJL++7+/+E+EV[I[+\( OOE " #O )N**28R8RVW8W !
#C(9(9DDJ
%
%
- \N!DJJ4::,a D U R b0 U R R [ R " S5 5 ggg)zCloses the progress bar.Nr/ )r, r3 rK rZ r[ r\ s r rH NotebookProgressBar.close s7 ;; 4;;#: KK tyy} - $; r )rG r2 r@ rC r" r r> r1 r+ r3 r, rA r r= r? r r6 r rD r )NTN, )FNN)__name__
__module__ __qualname____firstlineno____doc__rB r6 r r r boolr8 rK rE rT rH __static_attributes__r r r r( r( = s " H FL
!%6:
$$
$ $
23 $
$,1\C 1\t 1\hsm 1\f& :.r r( c L ^ \ rS rSrSrS U 4S jjrS rS rS
S jrS r Sr
U =r $ ) r- a
An object tracking the updates of an ongoing training with progress bars and a nice table reporting metrics.
Args:
num_steps (`int`): The number of steps during training. column_names (`List[str]`, *optional*):
The list of column names for the metrics table (will be inferred from the first call to
[`~utils.notebook.NotebookTrainingTracker.write_line`] if not set).
c L > [ TU ] U5 Uc S OU/U l S U l g ra )superr8 inner_table child_bar)r7 num_steps column_names __class__s r r8 NotebookTrainingTracker.__init__ s(
##/#74l^ r c d [ U R U R U R U R U R
5 U l U R b( U =R [ U R 5 -
sl U R b) U =R U R R -
sl U R c9 [ R " [ R " U R 5 SS9U l
g U R R [ R " U R 5 5 g rX )r r r r r r r" rm r& rn r3 rZ rT r[ rK r\ s r rT NotebookTrainingTracker.display s *4::tzz4;;PTPZPZ\`\f\fg ' NN 01A1A B BN >> % NNdnn 6 6 6N ;; ,,tyy '@TRDK KK tyy 8 9r c F U R c8 [ UR 5 5 [ UR 5 5 /U l gU R S nUR 5 H nX2; d M
UR U5 M X R S' [ U R 5 S: a U R S nU R S S nUS X :w a6 U R R U Vs/ s H ofU; a X OSPM sn5 gUnU H- nXgR 5 ; d M XBR
U5 Xv' M/ U Vs/ s H ogU PM snU R S' gU R R U Vs/ s H oaU PM sn5 gs snf s snf s snf )ze
Write the values in the inner table.
Args:
values (`Dict[str, float]`): The values to display.
Nr r zNo Log)rm listkeysvaluesappendrS index)r7 ry columnskey last_values first_columnc
new_valuess r
write_line"NotebookTrainingTracker.write_line sx # $V[[] 3T&--/5J KD &&q)G {{}%NN3' % #* Q 4##$q( " . .r 2 # / / 21 5 q>V%99$$++]d,e]dXYv+VY8-S]d,ef "(J $ OO$55,7
a8H,IJM % DK+K7aqM7+KD$$R( ''G(DGqG(DE -f ,L(Ds 0FF>F c 8 [ XXS9U l U R $ )a
Add a child progress bar displayed under the table of metrics. The child progress bar is returned (so it can be
easily updated).
Args:
total (`int`): The number of iterations for the child progress bar.
prefix (`str`, *optional*): A prefix to write on the left of the progress bar.
width (`int`, *optional*, defaults to 300): The width (in pixels) of the progress bar.
)r r, r )r( rn )r7 r r r s r add_child!NotebookTrainingTracker.add_child s -U$\~~ r c 2 SU l U R 5 g)z
Closes the child progress bar.
N)rn rT r\ s r remove_child$NotebookTrainingTracker.remove_child s r )rn r" rm r3 ra )Nr` ) rb rc rd re rf r8 rT r r r rh
__classcell__)rq s @r r- r- s'
: F> r r- c T \ rS rSrSrS rS rS rS
S jrS r S
S jr
S
S
jr S r S r
g)NotebookProgressCallbacki z~
A [`TrainerCallback`] that displays the progress of training or evaluation, optimized for Jupyter Notebooks or
Google colab.
c . S U l S U l SU l g )NF)training_trackerprediction_bar_force_next_updater\ s r r8 !NotebookProgressCallback.__init__& s $ ""' r c $ UR [ R :X a SOSU l SU l SU l U R /S/- nUR [ R :w a UR S5 [ UR U5 U l
g )NEpochStepr
Training LossValidation Loss)
eval_strategyr EPOCHr
training_losslast_logNOrz r- max_stepsr )r7 argsstatecontrolkwargsrp s r on_train_begin'NotebookProgressCallback.on_train_begin+ s} '+'9'9=M=S=S'SGY_
))*o->> !1!4!4 4 1 2 7 V r c " [ UR 5 UR :X a [ UR 5 O
UR S nU R R UR S- SU SUR
3U R S9 SU l g )NrQ r Epoch rP r2 r: F)r epochr rK global_stepnum_train_epochsr )r7 r r r r r s r on_step_end$NotebookProgressCallback.on_step_end4 s $' $4 $CEKK EKKX[K\ $$ ! UG1U%;%;$<= 00 %
#( r Nc | [ U5 ( d g U R cl U R b* U R R [ U5 5 U l O[ [ U5 5 U l U R R
S5 g U R R
U R R S- 5 g )Nr )r r r r rS r( rK r )r7 r r r eval_dataloaderr s r on_prediction_step+NotebookProgressCallback.on_prediction_step= s /** &$$0&*&;&;&E&Ec/FZ&[#&9#o:N&O# & &q ) & &t':':'@'@1'D Er c ` U R b U R R 5 S U l g ra )r rH r7 r r r r s r
on_predict#NotebookProgressCallback.on_predictI s( * % % ' "r c UR [ R :X a9 SU; a2 SUS 0nUR US' U R R U5 g g g )Nlossr r )r r r r r r )r7 r r r logsr ry s r on_log NotebookProgressCallback.on_logN sW !1!4!4 44 %tF|4F " . .F6N ! ! , ,V 4 :H 4r c U R Gb SSS.n[ UR 5 H nSU; d M US US' O U R S:X a [ UR
5 US' OUR US' SnU H3 n U R S5 ( d M [ R " S S
U 5 nM5 UR S S 5 n
UR S S 5 n
UR U S
3S 5 n
UR U S3S 5 n
UR U S3S 5 n
UR U S3S 5 n
UR 5 HT u pU R S5 n SR U SS V
s/ s H oR 5 PM sn
5 nUS:X a SnXU' MV U R R U5 U R R! 5 S U l SU l g g s sn
f )NzNo log)r r r r r r eval_lossz\_loss$r/
total_flosr _runtime_samples_per_second_steps_per_second_jit_compilation_time_rN r Lossr T)r reversed log_historyr r r r endswithresubpopr! splitjoin
capitalizer r r r )r7 r r r metricsr ry logmetric_key_prefixkr vsplitspartnames r on_evaluate$NotebookProgressCallback.on_evaluateV s ,'/HMF 1 12S=.1&kF?+ 3
G+"%ekk"2w!&!2!2v & ::g&&(*z2q(A% L$/A GT*A 0 1 :DAA 0 11D EtLA 0 11B CTJA 0 11F GNA
xxvabz Jzt !2z JK6>,D t
(
! ! , ,V 4 ! ! . . 0"&D &*D #A -. !Ks #G
c U R R UR S[ UR 5 SUR
3SS9 S U l g )Nr rP Tr )r rK r r r r r s r on_train_end%NotebookProgressCallback.on_train_endy sR $$ S - .a0F0F/GH %
!% r )r r r r r r ra )rb rc rd re rf r8 r r r r r r r rh r r r r r s2
(
W(
F#
5!+F%r r )r` )r4 r r< typingr IPython.displayrT rZ trainer_callbackr
trainer_utilsr r r r r& r( r- r r r r r sW
. 8D"R. R.jK 1 K\_% _%r