Z j@ S SK r S SKrS SKrS SKJrJr S SKJr SSK J
r
SSK J r J
r
\" S5 rS\S- S\S \4S
jrS rSS jrS
r " S S5 r " S S\5 r " S S\
5 rg) N)OptionalTypeVar )TrainerCallback)IntervalStrategy
has_length_Txmsgreturnc " U c [ U5 eU $ N) RuntimeError)r
r s l/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/utils/notebook.py_requirer s y3 H c 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 r format_timer # s[ AA4i!r'R R!A%&!VaS!C!C !CAc7!Ac71CCr 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/ 5 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 r; r: time
start_time last_time start_value elapsed_timepredicted_remainingwarmup first_callswait_for
update_barminr average_time_per_itemr4 closemaxr r? )r@ r rC r; 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.Nr8 )r5 r< rT rc rd re s r rQ NotebookProgressBar.close s7 ;; 4;;#: KK tyy} - $; r )rP r; rI rL r+ r" rG r: r4 r< r5 rJ r! rF rH r r? r rM r# )NTN, )FNr )__name__
__module__ __qualname____firstlineno____doc__rK r? r r) boolr rA rT rN r] rQ __static_attributes__r r r r1 r1 F s " H FL
"6:
$$ d
$ $
23 $
$,1\C 1\t 1\cDj 1\f& :.r r1 c L ^ \ rS rSrSrS U 4S jjrS rS rS
S jrS r Sr
U =r $ ) r6 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 r )superrA inner_table child_bar)r@ num_steps column_names __class__s r rA 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 ra )r$ r r r! r" r# r+ ru r/ rv r< rc r] rd rT re s r r] NotebookTrainingTracker.display s *4::tzz4;;PTPZPZ\`\f\fg ' NN 01A1A B BN >> % NNdnn 6 6 6N ;; ,,tyy '@TRDK KK tyy 8 9r c U R c8 [ UR 5 5 [ UR 5 5 /U l gU R S nU 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 a< U R R U Vs/ s H oaR
US5 PM sn5 gUnU H nXg; 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)ru listkeysvaluesappendr\ getindex)r@ r columnskey last_values first_columnc
new_valuess r
write_line"NotebookTrainingTracker.write_line se # $V[[] 3T&--/5J KD &&q)G %NN3' #* Q 4##$q( " . .r 2 # / / 21 5 q>V%99$$++g,VgZZ8-Dg,VW "(J $ .,7
a8H,IJM % DK+K7aqM7+KD$$R( ''G(DGqG(DE -W ,L(Ds "E>0F(Fc 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! r5 r# )r1 rv )r@ 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)rv r] re s r remove_child$NotebookTrainingTracker.remove_child! s r )rv r+ ru r< r )Nri ) rj rk rl rm rn rA r] r r r rp
__classcell__)ry s @r r6 r6 s'
: F> r r6 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_updatere s r rA !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_logNOr r6 max_stepsr )r@ argsstatecontrolkwargsrx s r on_train_begin'NotebookProgressCallback.on_train_begin4 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 n[ U R S5 nUR UR
S- SU SUR 3U R S9 SU l g )NrZ z0on_train_begin must be called before on_step_endr& Epoch rY r; rC F)r epochr r rT global_stepnum_train_epochsr )r@ r r r r r tts r on_step_end$NotebookProgressCallback.on_step_end= s $' $4 $CEKK EKKX[K\
d++-_
`
! 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 r\ r1 rT r )r@ r r r eval_dataloaderr s r on_prediction_step+NotebookProgressCallback.on_prediction_stepG 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 r )r rQ )r@ r r r r s r
on_predict#NotebookProgressCallback.on_predictS s( * % % ' "r c UR [ R :X aE SU; a> [ U R S5 nSUS 0nUR
US' UR
U5 g g g )Nlossz+on_train_begin must be called before on_logr r )r r r r r r r )r@ r r r logsr r r s r on_log NotebookProgressCallback.on_logX sb !1!4!4 44 $//1^_B%tF|4F " . .F6N MM& ! :H 4r c UR [ R :X a SOSU l SSS.n[ UR
5 H nSU; d M US US' O U R S:X a [
UR 5 US' OUR US' Uc 0 nSnU 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" b5 U R" nUR% U5 UR' 5 SU l Oc[* R, " [* R. " [1 [3 UR5 5 5 [3 UR7 5 5 /5 5 5 S U l g s sn
f )Nr r zNo log)r r r r eval_lossz\_loss$r8
total_flosr _runtime_samples_per_second_steps_per_second_model_preparation_time_rW r& Lossr T) r r r r reversed log_historyr r r endswithresubpopr* splitjoin
capitalizer r r r rc r] rd r/ r r r r )r@ r r r metricsr r logmetric_key_prefixkr vsplitspartnamer s r on_evaluate$NotebookProgressCallback.on_evaluatea s+ (,'9'9=M=S=S'SGY_#+IE - -.C}*-f+' /
' !%++ .F7O " . .F6N ?G "Azz'""$&FF:r1$=!
KK d + KK & KK,-X6 = KK,--@A4 H KK,-->? F KK,--DEt LMMODAWWS\F886!": F:4__ .: FGDv~( 4L
$ ,&&B MM& ! OO &*D #
LL#5tFKKM7JDQWQ^Q^Q`La6b#cd e "! Gs I
c [ U R S5 nUR UR S[ UR
5 SUR 3SS9 S U l g )Nz1on_train_begin must be called before on_train_endr rY Tr )r r rT r r r r )r@ r r r r r s r on_train_end%NotebookProgressCallback.on_train_end s]
d++-`
a
S - .a0F0F/GH
!% r )r r r r r r r )rj rk rl rm rn rA r r r r r r r rp r r r r r ) s2
(
W(
F#
".#`%r r )ri )r= r rE typingr r IPython.displayr] rc trainer_callbackr
trainer_utilsr r r r) r r r$ r/ r1 r6 r r r r r s
$ . 8
T]
T
D"R. R.jK 1 K\o% o%r