4888cb7a5299f7eed9214ddc70871dc05ea0f4d2,model/ft_ResNet50/train.py,,,#,298

Before Change

    base_params = filter(lambda p: id(p) not in ignored_params, model.parameters())
    optimizer_ft = optim.SGD([
             {"params": base_params, "lr": 0.1*opt.lr},
             {"params": model.model.fc.parameters(), "lr": opt.lr},
             {"params": model.classifier.parameters(), "lr": opt.lr}
         ], weight_decay=5e-4, momentum=0.9, nesterov=True)
else:

After Change

if opt.PCB:
    model = PCB(len(class_names))

opt.nclasses = len(class_names)

print(model)

if not opt.PCB:

Instances