a ;ƒdçã@s*ddlZddlmZGdd„dejƒZdS)éNcs$eZdZ‡fdd„Zdd„Z‡ZS)ÚBertClassificationModelcsHtt|ƒ ¡d}tj |¡|_|j ¡D] }d|_q*t   dd¡|_ dS)Nzbert-base-chineseTié) ÚsuperrÚ__init__Ú transformersÚ BertModelÚfrom_pretrainedÚbertÚ parametersÚ requires_gradÚnnÚLinearÚdense)ÚselfZpretrained_weightsÚparam©Ú __class__©ús