| z | z
|---|
| z | z
)
isinstancefloatstr)items html_codeilineeltr r r text_to_html_table, s
r" c @ s~ e Zd ZdZdZdZ ddedee d e d
ed d ef
d
dZ
d dede dee 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__e s
z NotebookProgressBar.__init__Fr force_updater+ 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 r+ r* time
start_time last_time start_value elapsed_timepredicted_remainingwarmup first_callswait_for
update_barminr average_time_per_itemr% closemaxr
r/ )r0 r r2 r+ 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 r7 r r8 r r> r+ display)r0 r r+ spaced_valuer r r r< 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& rI r, dispHTMLrB r0 r r r rI s
zNotebookProgressBar.displayc C s2 | j du r| jdur| jtd dS dS dS )zCloses the progress bar.Nr( )r& r, rB rM rN rO r r r r? s zNotebookProgressBar.close)NTNr )FNN)__name__
__module__ __qualname____doc__r9 r/ r
r r boolr1 rB r< rI 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 r d n|g| _d | _d S rP )superr1 inner_table child_bar)r0 num_steps column_names __class__r r r1 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 rK )r r r r r r r rW r" rX r, rM rI rN rB rO r r r rI s
z NotebookTrainingTracker.displayc s | j du rt t g| _ dS | j d } D ] }||vr(|| q || j d< t| j dkr|| j d }| j d d }|d | krX| j fdd|D dS |D ]}| vrm||| |< q\ 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 | ] }| v r | nd qS )zNo Logr .0cvaluesr r