a d @sjddlmZddlmZmZddlZddlmZddlm Z e dddZ Gd d d Z Gd d d Z dS) )Path)CallableOptionalN) TrainerState)returncCs d|dS)NZmodel_z.pt)xrr[/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/x/callbacks.pydefault_filename_fcn sr c@s:eZdZdZd eeeddddZej ee dd d Z dS) EarlyStoppingaCallback that evaluates whether there is improvement in the loss function. The module track the losses and in case of finish patience sends a termination signal to the trainer. Args: monitor: the name of the value to track. min_delta: the minimum difference between losses to increase the patience counter. patience: the number of times to wait until the trainer does not terminate. **Usage example:** .. code:: python early_stop = EarlyStopping( monitor="top5", filepath="early_stop_model.pt" ) trainer = ImageClassifierTrainer(..., callbacks={"on_checkpoint", early_stop} ) N)monitor min_deltapatiencercCs(||_||_||_d|_d|_d|_dS)NrF)rrrcounter best_score early_stop)selfrrrrrr __init__&s zEarlyStopping.__init__modelepochrcCs~||jj}|jdur||_n>||j|jkrP|jd7_|j|jkr\d|_n ||_d|_|jrxtd|dtj Stj S)NrTrz3[INFO] Early-Stopping the training process. Epoch: .) ravgrrrrrprintrZ TERMINATEZTRAINING)rrr valid_metricZscorerrr __call__/s   zEarlyStopping.__call__)r r) __name__ __module__ __qualname____doc__strfloatintrnnModulerrrrrr r s r c@s>eZdZdZd eeeeddddZej e ddddZ dS) ModelCheckpointaCallback that save the model at the end of every epoch. Args: filepath: the where to save the mode. monitor: the name of the value to track. **Usage example:** .. code:: python model_checkpoint = ModelCheckpoint( filepath="./outputs", monitor="top5", ) trainer = ImageClassifierTrainer(..., callbacks={"on_checkpoint", model_checkpoint} ) N)filepathr filename_fcnrcCs4||_||_|pt|_d|_t|jjddddS)Nr T)parentsexist_ok)r)rr _filename_fcn best_metricrmkdir)rr)rr*rrr rWs  zModelCheckpoint.__init__rcCs@||jj}||jkr<||_t|j||}t||dS)N)rrr.rr)r-torchsave)rrrrZvalid_metric_valuefilenamerrr rbs   zModelCheckpoint.__call__)N) rr r!r"r#rrrr&r'r%rrrrr r(Cs r()pathlibrtypingrrr0Ztorch.nnr&utilsrr#r r r(rrrr s   4