David310/Detect_AI-generated_Image
4
1U
2 �`jf� � @ s" d dl Zd dlZG dd� d�ZdS )� Nc @ s* e Zd ZdZd dd�Zdd� Zd d3� Zd S )
�
EarlyStoppingzSEarly stops the training if validation loss doesn't improve after a given patience.� Fr c C s2 || _ || _d| _d| _d| _tj | _|| _dS )a� 4 Args:5 patience (int): How long to wait after last time validation loss improved.6 Default: 77 verbose (bool): If True, prints a message for each validation loss improvement. 8 Default: False9 delta (float): Minimum change in the monitored quantity to qualify as an improvement.10 Default: 011 r NF) �patience�verbose�counter�12best_score�13early_stop�np�Inf� score_max�delta)�selfr r r � r �1/root/autodl-tmp/UniversalFakeDetect/earlystop.py�__init__ s 1415zEarlyStopping.__init__c C s� | j d kr || _ | �||� nb|| j | j k rh| jd7 _td| j� d| j� �� | j| jkr�d| _n|| _ | �||� d| _d S )Nr zEarlyStopping counter: z out of Tr )r �save_checkpointr r �printr r �r
�score�modelr r r �__call__ s 16 zEarlyStopping.__call__c C s6 | j r"td| jd�d|d�d�� |�d� || _dS )z*Saves model when validation loss decrease.z Validation accuracy increased (z.6fz --> z). Saving model ...�bestN)r r r �
save_networksr r r r r ' s 17z EarlyStopping.save_checkpointN)r Fr )�__name__�18__module__� __qualname__�__doc__r r r r r r r r s 19r )�numpyr �torchr r r r r �<module> s