/usr/local/lib/python3.6/site-packages/transformers/utils/__pycache__/notebook.cpython-36.pyc (12190B)
| 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?NT, 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 s| || _ |dk r|| _| jdkrbtj | _| _| | _| _d | _| _| j | _
d| _ | j | n|| jkrv| rvdS |s| j
dks|t
| j| j | jkrx| j
dkr| j
d8 _
tj }|| j | _|| jkr| j|| j | _nd| _|| jkr| j}d| _| js6| j n | jdk r6| j| 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 _| j d S ) N [/z : < :z < z, r z.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| jj d S | j d krXt
jt
j | jdd| _ n| j j t
j | j d S )NT)
display_id)
r r r r r r r r' rB r+ dispHTMLr< )r, r r r rB s
zNotebookProgressBar.displayc C s* | j dkr&| jdk r&| jjtjd dS )zCloses the progress bar.Nr( )r' r+ r< rD rE )r, r r r r8 s zNotebookProgressBar.close)NTNr$ )FN)N)__name__
__module__ __qualname____doc__r3 r: r r r boolr- r< r5 rB r8 r r r r r" = s # 4
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 j| |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 j
| jdd| _
n| j
jt j
| j d S )NT)rC )r r r r r r r rL r! rM r+ rD rB rE r< )r, r r r rB s
z NotebookTrainingTracker.displayc s | j dkr&t j t j g| _ n`| j d }t| j dkrlx" j D ]}||krH|j| qHW || j d< | j 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 | ] } | qS r r ).0c)valuesr r