o ¡¿¯iã@s4ddlZddlZddlZddlZdd„Zdd„ZdS)éNcCslt |j¡dtjj_dtjj_|jdkrdtj _ t   d¡|jdkr4dtjj_dtjj_t   d¡dSdS)zˆEnsures pytorch produces deterministic results depending on the program arguments :param Namespace args: The program arguments TFéztorch type check is disabledéz%torch cudnn deterministic is disabledN) ÚtorchÚ manual_seedÚseedÚbackendsÚcudnnÚ deterministicÚ benchmarkÚ debugmodeÚchainerÚconfigÚ type_checkÚloggingÚinfo©Úargs©rúT/home/ubuntu/.local/lib/python3.10/site-packages/espnet/utils/deterministic_utils.pyÚset_deterministic_pytorchs  ÿ     ýrcCslt|jƒtjd<t dtjd¡|jdkr dtj_ t d¡|jdkr0dtj_ t d¡d Sdtj_ d S) zˆEnsures chainer produces deterministic results depending on the program arguments :param Namespace args: The program arguments Ú CHAINER_SEEDzchainer seed = rFzchainer type check is disabledrz'chainer cudnn deterministic is disabledTN) ÚstrrÚosÚenvironrrr r r rÚcudnn_deterministicrrrrÚset_deterministic_chainer"s    r)rrr rrrrrrrÚs