Pytorch-yolov3: how to change the learning-rate？

[W998K]

how to change the learning-rate？

[jaysonph]

@W998K
Before training, you can simply set like:
optimizer = torch.optim.Adam(model.parameters(), lr=0.0001)

Within training, I suggest you can try LR scheduler (very useful for training):

optimizer = torch.optim.SGD(model.parameters(), lr=0.1, momentum=0.9)
scheduler = ReduceLROnPlateau(optimizer, 'min')
for epoch in range(10):
    train(...)
    val_loss = validate(...)
    # Note that step should be called after validate()
    scheduler.step(val_loss)