fujiyuu75
4/8/2020 - 9:32 AM

how to freeze layers in pytorch.

def freeze_network(model):
    for name, p in model.named_parameters():
        if "l0" not in name:
            p.requires_grad = False
    return model

---

model = freeze_network(model)