/usr/local/lib/python3.6/site-packages/tqdm/__pycache__
Edit: /usr/local/lib/python3.6/site-packages/tqdm/__pycache__/keras.cpython-36.pyc (4904B)
3
Eg9 @ s d dl mZmZ d dlmZ d dlmZ ddlmZ yd dl Z W nP e
efk
r Z z0yd dl
m Z W n e
k
r~ eY nX W Y ddZ[X nX dd giZd
gZG dd
d
e jjZdS ) )absolute_importdivision)copy)partial )tqdmN)keraszgithub.com/Z casperdclTqdmCallbackc @ s~ e Zd ZdZeddd fddZddddefdd Zd
d Zdd
Z dd Z
dd Zedd Zedd Z
edd ZdS )r z,Keras callback for epoch and batch progress.Nc C s dS )Nr )logsr
r
4/usr/local/lib/python3.6/site-packages/tqdm/keras.py
s zTqdmCallback.c s d fdd }|S )Nc sF } r8r*t fddD j dd j| d S )Nc s g | ]} j |d qS )r )pop).0i)r r
r
s z?TqdmCallback.bar2callback..callback..F)Zrefresh)r Zset_postfixupdate)_r n)bardeltar )r r callback s z+TqdmCallback.bar2callback..callback)Nr
)r r r r r
)r r r r bar2callback s zTqdmCallback.bar2callbackr c K s |rt |f|}|| _||dd| _| j| j| _|rR|rR|| d | | _}n
d | _}|| _|dkr||ddd| _| j| jddgd d
d| _dS )a^
Parameters
----------
epochs : int, optional
data_size : int, optional
Number of training pairs.
batch_size : int, optional
Number of training pairs per batch.
verbose : int
0: epoch, 1: batch (transient), 2: batch. [default: 1].
Will be set to `0` unless both `data_size` and `batch_size`
are given.
tqdm_class : optional
`tqdm` class to use for bars [default: `tqdm.auto.tqdm`].
tqdm_kwargs : optional
Any other arguments used for all bars.
epoch)totalunitr NbatchF)r r leavesizec S s | j ddS )Nr r )get)r r
r
r r
C s z'TqdmCallback.__init__..)r r ) r
tqdm_class epoch_barr Zon_epoch_endbatchesverbose batch_baron_batch_end)selfepochsZ data_size
batch_sizer# r Ztqdm_kwargsr" r
r
r __init__" s
zTqdmCallback.__init__c O s> | j j}|d|dd }|d k r:|| jjkr:| jj|d d S )Nr' Znb_epoch)r )paramsr r! r reset)r&