o i@s&ddlZddlZddZdddZdS)NcCs8|jdj}||dkrtdt|ddSdS)zChecks an early stopping trigger and warns the user if it's the case :param trainer: The trainer used for training :param epochs: The maximum number of epochs mainzHit early stop at epoch z= You can change the patience or set it to 0 to run all epochsN)updater get_iteratorepochloggingwarningstr)trainerepochs end_epochr U/home/ubuntu/.local/lib/python3.10/site-packages/espnet/utils/training/train_utils.pycheck_early_stops rFcCsX|j}|j}|r |jn|j}d|vrdnd}|dkr*tjjj||||dfd|_dSdS)zSets the early stop trigger given the program arguments :param trainer: The trainer used for training :param args: The program arguments :param is_lm: If the trainer is for a LM (epoch instead of epochs) accmaxminrr)monitormodepatients max_triggerN) patienceearly_stop_criterionrr chainertrainingtriggersEarlyStoppingTrigger stop_trigger)r argsis_lmr criterionr rr r rset_early_stops r!)F)rrrr!r r r rs