2、指圆其次当strict = True时,要求加载的权重文件中的参数与模型中的完全匹配,设置strict = False时,即可使用权重文件中与模型匹配层的数据,其余的默认初始化。
3、最后当模型更改后不能如上述情况加载预训练权重时,可以考虑获取原模型的权重加载到新模型上。
左右权重文件的产生,是根腊闷则据模型训练的过程而产生的。首先,根据训练数据,建立模型,定义模型的参数;然后,使用梯度轮棚下降法,不断调整参数,使模型的损失函数最小;最后,收敛到最优解,得到模型的权重参数,罩凳即权重文件。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)