L i>ddlZddlZdZejdZy)Nc"t5tjd||i|}t|tjr7|j j dk(rtjj|cdddS#1swYyxYw)zWrapper to set seed manually for some functions like dropout See: https://github.com/pytorch/pytorch/pull/62315#issuecomment-896143189 for more details. *lazyN) freeze_rng_statetorch manual_seed isinstanceTensordevicetype_lazy mark_step)opargskwargsoutputs Z/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/torch/testing/_utils.pywrapper_set_seedr sp    "T$V$ fell + 0B0Bf0L KK ! ! #   s A0BBc #Ktjjj5tjj 5tj }tjjrtjj }dddddd dtjjj5tjj 5tjjrtjjtjddddddy#1swYxYw#1swYxYw#1swY*xYw#1swYyxYw#tjjj5tjj 5tjjrtjjtjdddn #1swYnxYwdddw#1swYwxYwxYww)N) rutils _mode_utils no_dispatch_C_DisableFuncTorch get_rng_statecuda is_available set_rng_state) rng_statecuda_rng_states rrrs  , , .80J0J0L8'') :: " " $"ZZ557N88+ [[ $ $ 0 0 2 +EHH4N4N4P +zz&&( ((8    * + + +8888 + + + +U[[ $ $ 0 0 2 +EHH4N4N4P +zz&&( ((8    * + + + + + +s)IE' AEE'#I,F 0(IE?7AE3 E? IE$ E''E0,I3E< 8E??FI )I4IAH/ & I/H8 4I; II II) contextlibrrcontextmanagerrrr%s*    ++r$