o ei@s4dZddlZddlZddlZdZddZddZdS)zTThis module implements some workarounds for dependencies Authors * Aku Rouhe 2022 NWEAKREFcCs.|}|ddurt|d<t||dS)N _scale_fn_ref) state_dictgetWEAKREF_MARKERtorchsave)objpathrr \/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/speechbrain/utils/_workarounds.py_cycliclrsaversr c Cs~d}tj||d}|dtkr+t|jtjs+d}|d7}|d7}|d7}t |z|j tj||ddd WdSt yN| tj||dYdSw) Ncpu) map_locationrzKLoading CyclicLR scheduler and the _scale_ref_fn did not exist in instance.zB You did not construct it with the same parameters it was created!z+ Looks like you changed the scale function!zE If this was not intentional, the scheduler might not work correctly.T)strict) rloadrr isinstancerweakref WeakMethodwarningswarnload_state_dict TypeError)r r end_of_epochdevicerMSGr r r _cycliclrloaders  r)__doc__rrrrr rr r r r s