
    &Vf                     p    d dl Zd dlmZ d dlmZ d dlmZ  ed           G d de                      ZdS )    N)keras_export)Callback)io_utilszkeras.callbacks.TerminateOnNaNc                       e Zd ZdZddZdS )TerminateOnNaNzACallback that terminates training when a NaN loss is encountered.Nc                     |pi }|                     d          }|Nt          j        |          st          j        |          r(t	          j        d| d           d| j        _        d S d S d S )NlosszBatch z$: Invalid loss, terminating trainingT)getnpisnanisinfr   	print_msgmodelstop_training)selfbatchlogsr	   s       a/var/www/html/software/conda/lib/python3.11/site-packages/keras/src/callbacks/terminate_on_nan.pyon_batch_endzTerminateOnNaN.on_batch_end   s    zrxxx~~ 0$ 0"HUHHH   ,0
((( 0 0    )N)__name__
__module____qualname____doc__r    r   r   r   r      s.        KK0 0 0 0 0 0r   r   )	numpyr   keras.src.api_exportr   keras.src.callbacks.callbackr   keras.src.utilsr   r   r   r   r   <module>r       s        - - - - - - 1 1 1 1 1 1 $ $ $ $ $ $ .//0 0 0 0 0X 0 0 0/0 0 0r   