o
    2h                     @   sH   d dl Zd dlmZ d dlmZ d dlmZ edG dd deZdS )    N)keras_export)Callback)io_utilszkeras.callbacks.TerminateOnNaNc                   @   s   e Zd ZdZdddZdS )TerminateOnNaNzACallback that terminates training when a NaN loss is encountered.Nc                 C   sT   |pi }| d}|d ur&t|st|r(t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    r   _/var/www/html/chatgem/venv/lib/python3.10/site-packages/keras/src/callbacks/terminate_on_nan.pyon_batch_end   s   

zTerminateOnNaN.on_batch_end)N)__name__
__module____qualname____doc__r   r   r   r   r   r      s    r   )	numpyr   keras.src.api_exportr   keras.src.callbacks.callbackr   keras.src.utilsr   r   r   r   r   r   <module>   s    