U Māhöć@s*ddlZddlZdd„Zejdd„ƒZdS)éNc OsXtƒHt d”|||Ž}t|tjƒr>|jjdkr>tj ”|W5QR£SQRXdS)z”Wrapper to set seed manually for some functions like dropout See: https://github.com/pytorch/pytorch/pull/62315#issuecomment-896143189 for more details. é*ZlazyN) Śfreeze_rng_stateŚtorchZ manual_seedŚ isinstanceZTensorZdeviceŚtypeZ_lazyZ mark_step)ŚopŚargsŚkwargsŚoutput©r śF/var/www/html/venv/lib/python3.8/site-packages/torch/testing/_utils.pyŚwrapper_set_seed s    r ccsŖtjj ”8tj ”"t ”}tj ”r6tj ”}W5QRXW5QRXz dVW5tjj ”<tj ”&tj ”r†tj  |”t  |”W5QRXW5QRXXdS)N) rŚutilsZ _mode_utilsZ no_dispatchZ_CZ_DisableFuncTorchZ get_rng_stateŚcudaZ is_availableZ set_rng_state)Z rng_stateZcuda_rng_stater r r rs    r)Ś contextlibrr Ścontextmanagerrr r r r Śs