| z | z
|---|
| z | z
)
isinstancefloatstr)items html_codeilineeltr r r text_to_html_table, s
r" c @ sf e Zd ZdZdZdZdeee e ed edd d
Z
dee ed 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 krdn|| _|| _|| _|| _d | _d | _d | _d S )N )r r r& r' r
last_valuecommentoutput)selfr r r& r' r r r r __init__e s z NotebookProgressBar.__init__F)r force_updater* c C sz || _ |dk r|| _| jdkrbt | _| _| | _| _d | _| _| j | _
d| _ | | n|| jkrt|stdS |s| j
dks|t
| j| j | jkrv| j
dkr| j
d8 _
t }|| j | _|| jkr| j|| j | _nd| _|| jkr| j}d| _| js4| n | jdk r4| j| j| | _| | || _|| _| jdkr^d| _ 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
update_every)r, r r. r* current_timer r r updatev sB
(
zNotebookProgressBar.updatec C s dt t| jt t| t| }| jd krHd| d| j d| _nv| jd krtd| d| j dt| j | _nJd| d| j dt| j dt| j | _| jdd| j dd 7 _| j| jd kst | jd
krd n d| j d 7 _| d S ) N [/z : < :z < z, r .2fz it/sr ])
lenr r r1 r r2 r r7 r* display)r, r r* Z spaced_valuer r r r5 s &
"( 4z NotebookProgressBar.update_barc C sp t | j| j| j| j| j| _| jd k r4| j d S | j d krXt
jt
| jdd| _ n| j t
| j d S NT)Z
display_id)
r r r r r r r r' rC r+ dispHTMLr< r, r r r rC s
zNotebookProgressBar.displayc C s* | j dkr&| jdk r&| jtd dS )zCloses the progress bar.Nr( )r' r+ r< rE rF rG r r r r8 s zNotebookProgressBar.close)NTNr )FN)N)__name__
__module__ __qualname____doc__r3 r: r
r r boolr- r< r5 rC r8 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 krd n|g| _d | _d S N)superr- inner_table child_bar)r, Z num_steps column_names __class__r r r- s z NotebookTrainingTracker.__init__c C s t | j| j| j| j| j| _| jd k r:| jt| j7 _| j d k rV| j| j j7 _| j
d krzt j t
| jdd| _
n| j
t
| j d S rD )r r r r r r r rO r" rP r+ rE rC rF r< rG r r r rC s
z NotebookTrainingTracker.displayc s | j dkr&t t g| _ n| j d } D ]}||kr8|| q8|| j d< t| j dkr| j d }| j d d }|d | kr| j fdd|D nD |D ]"}| kr||| |< 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 | ]}| kr | nd qS )zNo Logr .0cvaluesr r