| z | z
|---|
| z | z
)
isinstancefloatstr)items html_codeilineeltr r r text_to_html_table, s
r" c @ sj e Zd ZdZdZdZdeee e ed edd d
Z
dee ee d d
dZ dddZ 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 NotebookTrainingTracker)r r leaveparentr c C s\ || _ |d u rdn|| _|| _|| _|| _d | _d | _d | _d | _d | _ dt
j v rXd| _ d S )N Z
VSCODE_PIDg ?)
r r r& r' r
last_valuecommentoutputr r osenviron update_every)selfr r r& r' r r r r __init__e s
z NotebookProgressBar.__init__F)r force_updater* c C s || _ |dur|| _| jdu rbt | _| _| | _| _d | _| _| j | _
d| _ | | n || jkrt|stdS |s| j
dks|t
| j| j | jkr| j
dkr| j
d8 _
t }|| j | _|| jkr| j|| j | _nd| _|| jkr| j}d| _| js4| n | jdur4| j| j| | _| | || _|| _| jdu sb| jdkrjd| _ ntt| j| j d| _ 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 r* r) time
start_timeZ last_timeZ start_value elapsed_timepredicted_remainingwarmupZ first_callswait_for
update_barminr average_time_per_itemr& closemaxr
r. )r/ r r1 r* current_timer r r update{ sB
(
zNotebookProgressBar.updatec C s dt t| jt t| t| }| jd u rHd| d| j d| _n| jd u rtd| d| j dt| j | _ndd| d| j dt| j dt| j | _| jdkr| jd7 _n | jdd | j d
d 7 _| j| jd u st | jdkrd n d| j d 7 _| d S )
N [/z : < :z < r z , +inf it/sz, r .2fz it/s])
lenr r r4 r r5 r r: r* display)r/ r r* Z spaced_valuer r r r8 s &
"
4z NotebookProgressBar.update_barc C sp t | j| j| j| j| j| _| jd ur4| j d S | j d u rXt
jt
| jdd| _ n| j t
| j d S NT)Z
display_id)
r r r r r r r r' rE r+ dispHTMLr> r/ r r r rE s
zNotebookProgressBar.displayc C s* | j du r&| jdur&| jtd dS )zCloses the progress bar.Nr( )r' r+ r> rG rH rI r r r r; s zNotebookProgressBar.close)NTNr )FN)N)__name__
__module__ __qualname____doc__r6 r. r
r r boolr0 r> r8 rE r; r r r r r# = 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 rd n|g| _d | _d S N)superr0 inner_table child_bar)r/ Z num_steps column_names __class__r r r0 s z NotebookTrainingTracker.__init__c C s t | j| j| j| j| j| _| jd ur:| jt| j7 _| j d urV| j| j j7 _| j
d u rzt j t
| jdd| _
n| j
t
| j d S rF )r r r r r r r rQ r" rR r+ rG rE rH r> rI r r r rE s
z NotebookTrainingTracker.displayc s | j du r&t t g| _ n| j d }D ]}||vr4|| q4|| j d< t| j dkr| j d }| j d d }|d | kr| j fdd|D n@ |D ] }| vr||| |< q fdd|D | j d< n| 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