23 lines
2.6 KiB
Plaintext
23 lines
2.6 KiB
Plaintext
|
3
|
|||
|
|
<EFBFBD><[0 <00> @ sD d dl mZmZ d dlZd dlZG dd<04> de<01>ZG dd<06> de<02>ZdS )<07> )<02> TensorBoard<72>ModelCheckpointNc s0 e Zd ZdZd<08> fdd<04> Zd <09> fdd<07> Z<05> ZS )
|
|||
|
|
<EFBFBD>CustomTensorBoardz& to log the loss after each batch
|
|||
|
|
<20> c s" t t| <00>jf |<02> || _d| _d S )Nr )<05>superr <00>__init__<5F> log_every<72>counter)<03>selfr <00>kwargs)<01> __class__<5F> <00>F/home/dlsaavedra/Desktop/Tesis/8.-YOLO/keras-yolo3-master/callbacks.pyr s zCustomTensorBoard.__init__Nc s<> | j d7 _ | j | j dkrzxP|j<02> D ]D\}}|dkr:q(tj<04> }|jj<06> }|j<07> |_||_ | j
|
|||
|
|
j || j <00> q(W | j
|
|||
|
|
j <0C> t
t| <00>j||<02> d S )Nr r <00>batch<63>size)r r )r r <00>items<6D>tf<74>Summary<72>value<75>add<64>item<65> simple_value<75>tag<61>writer<65> add_summary<72>flushr r <00> on_batch_end)r
|
|||
|
|
r <00>logs<67>namer <00>summary<72>
summary_value)r r
r r
s
|
|||
|
|
|
|||
|
|
|
|||
|
|
z CustomTensorBoard.on_batch_end)r )N)<07>__name__<5F>
|
|||
|
|
__module__<EFBFBD> __qualname__<5F>__doc__r r <00>
__classcell__r
r
)r r r s r c s. e Zd ZdZ<03> fdd<03>Zd<07> fdd<06> Z<05> ZS )<08>CustomModelCheckpointz9 to save the template model, not the multi-GPU model
|
|||
|
|
c s t t| <00>jf |<02> || _d S )N)r r& r <00>
model_to_save)r
|
|||
|
|
r' r )r r
r r s z CustomModelCheckpoint.__init__Nc s^ |pi }| j d7 _ | j | jk<05>rHd| _ | jjf d|d i|<02><02>}| jr<>|j| j<06>}|d krrtjd| j t <09> n<>| j
|
|||
|
|
|| j <0B>r<>| j dkr<>t
d|d | j| j ||f <00> || _ | jr<>| jj|dd<07> q<>| jj|dd<07> n$| j dkr<>t
d|d | j| j f <00> nJ| j dk<04>r t
d |d |f <00> | j<0E>r8| jj|dd<07> n| jj|dd<07> tt| <00>j||<02> d S )
|
|||
|
|
Nr r <00>epochz5Can save best model only with %s available, skipping.z@
|
|||
|
|
Epoch %05d: %s improved from %0.5f to %0.5f, saving model to %sT)<01> overwritez*
|
|||
|
|
Epoch %05d: %s did not improve from %0.5fz
|
|||
|
|
Epoch %05d: saving model to %s)<15>epochs_since_last_save<76>period<6F>filepath<74>format<61>save_best_only<6C>get<65>monitor<6F>warnings<67>warn<72>RuntimeWarning<6E>
|
|||
|
|
monitor_op<EFBFBD>best<73>verbose<73>print<6E>save_weights_onlyr' <00> save_weights<74>saver r& r )r
|
|||
|
|
r( r r, <00>current)r r
r <00> on_epoch_end# s8
|
|||
|
|
|
|||
|
|
z"CustomModelCheckpoint.on_epoch_end)N)r! r" r# r$ r r< r% r
r
)r r r& s r& ) <09>keras.callbacksr r <00>
|
|||
|
|
tensorflowr <00>numpy<70>npr r& r
r
r
r <00><module> s
|