Add parser argument to set LR milestone for training

akamaster / pytorch_resnet_cifar10

Proper implementation of ResNet-s for CIFAR10/100 in pytorch that matches description of the original paper.

BSD 2-Clause "Simplified" License

1.22k stars 335 forks source link

Add parser argument to set LR milestone for training #38

Closed JonnaMat closed 1 year ago

JonnaMat commented 1 year ago

When training with a different number of epochs it makes sense to adjust the LR schedule. See my comment in the discussion About the number of epochs?.