o –h˜ã@sddlZGdd„dƒZdS)éNc@s$eZdZdd„Zdd„Zdd„ZdS)ÚTrainercCs|jjD]}||d<qdS)NÚlr)Ú optimizerÚparam_groups)ÚselfÚnew_lrÚparam_group©r úW/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/models/common/trainer.pyÚ change_lrs ÿzTrainer.change_lrcCs&|j ¡|j ¡dœ}t ||¡dS)N)Úmodelr)rÚ state_dictrÚtorchÚsave©rÚfilenameÚsavedictr r r rsþzTrainer.savecCsJtj|dd„dd}|j |d¡|jddkr#|j |d¡dSdS) NcSs|S)Nr )ÚstorageÚlocr r r ÚszTrainer.load..T)Úweights_onlyrÚmodeÚtrainr)rÚloadrÚload_state_dictÚargsrrr r r rs ÿzTrainer.loadN)Ú__name__Ú __module__Ú__qualname__rrrr r r r rsr)rrr r r r Ús