其他分享
首页 > 其他分享> > 如何固化预训练模型的参数

如何固化预训练模型的参数

作者:互联网

# 自定义全连接层加载VGG16作为预训练模型class ModifiedVGG16Model(torch.nn.Module):
    def __init__(self):
        super(ModifiedVGG16Model, self).__init__()
        # 加载预训练模型

               

标签:__,模型,init,参数,固化,加载,self,ModifiedVGG16Model
来源: https://blog.51cto.com/u_15177056/2725768