| z | z
|---|
| z | z
)
isinstancefloatstr)items html_codeilineeltr r r text_to_html_table5 s
r* c @ s~ e Zd ZdZdZdZ ddededB d ed
e d d ef
d
dZ
d dedededB fddZ d ddZ dd Z
dd ZdS ) NotebookProgressBara
A progress par for display in a notebook.
Class attributes (overridden by derived classes)
- **warmup** (`int`) -- The number of iterations to do at the beginning while ignoring `update_every`.
- **update_every** (`float`) -- Since calling the time takes some time, we only do it every presumed
`update_every` seconds. The progress bar uses the average time passed up until now to guess the next value
for which it will call the update.
Args:
total (`int`):
The total number of iterations to reach.
prefix (`str`, *optional*):
A prefix to add before the progress bar.
leave (`bool`, *optional*, defaults to `True`):
Whether or not to leave the progress bar once it's completed. You can always call the
[`~utils.notebook.NotebookProgressBar.close`] method to make the bar disappear.
parent ([`~notebook.NotebookTrainingTracker`], *optional*):
A parent object (like [`~utils.notebook.NotebookTrainingTracker`]) that spawns progress bars and handle
their display. If set, the object passed must have a `display()` method.
width (`int`, *optional*, defaults to 300):
The width (in pixels) that the bar will take.
Example:
```python
import time
pbar = NotebookProgressBar(100)
for val in range(100):
pbar.update(val)
time.sleep(0.07)
pbar.update(100)
``` g?NTr r r leaveparentNotebookTrainingTrackerr c C s` || _ |d u r dn|| _|| _|| _|| _d | _d | _d | _d | _d | _ dt
j v r.d| _ d S d S )N
VSCODE_PIDg ?)
r r r- r. r
last_valuecommentoutputr r osenviron update_every)selfr r r- r. r r r r __init__n s
z NotebookProgressBar.__init__Fr force_updater3 c C s || _ |dur
|| _| jdu r1t | _| _| | _| _d | _| _| j | _
d| _ | | dS || jkr:|s:dS |sM| j
dksM|t
| j| j | jkr| j
dkrY| j
d8 _
t }|| j | _|| jkrr| j|| j | _nd| _|| jkr| j}d| _| js| n| jdur| j| j| | _| | || _|| _| jdu s| jdkrd| _ dS tt| j| j d| _ dS dS )a
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 r3 r2 time
start_time last_time start_value elapsed_timepredicted_remainingwarmup first_callswait_for
update_barminr average_time_per_itemr- closemaxr r7 )r8 r r: r3 current_timer r r update sF
&
zNotebookProgressBar.updatec C s dt t| jt t| t| }| jd u r$d| d| j d| _nH| jd u r:d| d| j dt| j | _n2d| d| j dt| j dt| j | _| jdkr]| jd7 _n| jdd | j d
d 7 _| j| jd u s{t | jdkr}d nd| j d 7 _| d S )
N [/z : < :z < r z , +inf it/sz, r! .2fz it/s])
lenr$ r r? r r@ r rF r3 display)r8 r r3 spaced_valuer r r rD s &
"
4 z NotebookProgressBar.update_barc C sr t | j| j| j| j| j| _| jd ur| j d S | j d u r-t
jt
| jdd| _ d S | j t
| j d S NT)
display_id)
r r r r r r r& r. rQ r4 dispHTMLrJ r8 r r r rQ s
zNotebookProgressBar.displayc C s2 | j du r| jdur| jtd dS dS dS )zCloses the progress bar.Nr0 )r. r4 rJ rU rV rW r r r rG s zNotebookProgressBar.close)NTNr )FNr )__name__
__module__ __qualname____doc__rA r7 r r$ boolr r9 rJ rD rQ rG r r r r r+ F s. $
3 r+ c sD e Zd ZdZd fdd Zdd Zdd Zdd
d Zd d
Z Z S )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).
Nc s* t | |d u r d n|g| _d | _d S r )superr9 inner_table child_bar)r8 num_steps column_names __class__r r r9 s
z NotebookTrainingTracker.__init__c C s t | j| j| j| j| j| _| jd ur | jt| j7 _| j d ur+| j| j j7 _| j
d u r>t j t
| jdd| _
d S | j
t
| j d S rS )r r r r r r r& r^ r* r_ r4 rU rQ rV rJ rW r r r rQ s
z NotebookTrainingTracker.displayc s | j du rt t g| _ dS | j d }D ] }||vr&|| q|| j d< t| j dkrx| j d }| j d d }|d | krV| j fdd|D dS |D ]}| vri||| |< qZ fdd|D | j d< dS | j fdd|D dS ) z
Write the values in the inner table.
Args:
values (`dict[str, float]`): The values to display.
Nr r! c s g | ]} |d qS )zNo Log)get.0cvaluesr r