首页 > 其他分享> > 如何固化预训练模型的参数 如何固化预训练模型的参数 2021-04-22 20:53:35 作者:互联网 # 自定义全连接层加载VGG16作为预训练模型class ModifiedVGG16Model(torch.nn.Module): def __init__(self): super(ModifiedVGG16Model, self).__init__() # 加载预训练模型 标签:__,模型,init,参数,固化,加载,self,ModifiedVGG16Model 来源: https://blog.51cto.com/u_15177056/2725768