a6d69cefe0727bc9d29c2308519a422c69f30739,train.py,,main,#Any#,30

Before Change

        // Set optimizer
        // optimizer = torch.optim.Adam(filter(lambda p: p.requires_grad, model.parameters()))
        optimizer = torch.optim.SGD(filter(lambda p: p.requires_grad, model.parameters()), lr=1e-3,
                                    momentum=.9, weight_decay=5e-4, nesterov=True)

        start_epoch = checkpoint["epoch"] + 1
        if checkpoint["optimizer"] is not None:
            optimizer.load_state_dict(checkpoint["optimizer"])

After Change

        //         p.requires_grad = False

        // Set optimizer
        optimizer = torch.optim.Adam(filter(lambda p: p.requires_grad, model.parameters()))
        // optimizer = torch.optim.SGD(filter(lambda p: p.requires_grad, model.parameters()), lr=1e-3, momentum=.9, weight_decay=5e-4, nesterov=True)

        start_epoch = checkpoint["epoch"] + 1
        if checkpoint["optimizer"] is not None:

Instances