pytorch冻结前k层

阿里云2000元红包!本站用户参与享受九折优惠!

优化器:

optimizer = torch.optim.SGD(filter(lambda p: p.requires_grad, network.parameters()),

                            lr=1e-3,

                            momentum=0.9

                            )

网络:

net = torch.hub.load(“facebookresearch/FixRes”,”fixresnet_50_CutMix”)

    net.fc = nn.Linear(net.fc.in_features,8)

    count = 0

    for child in net.children():

        count+=1

        if count <= 4:

            print(child)

            for param in child.parameters():

                param.requires_grad = False

https://www.jianshu.com/p/1f4c12e5abc7

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论