the usage of the 'orch.optim.lr_scheduler'

[ccijunk]

line 90 , in main.py , the usage of the 'orch.optim.lr_scheduler' might be wrong.That code couldn't adjust the lr

the example in Document is as the follow

 Example:
        >>> # Assuming optimizer uses lr = 0.05 for all groups
        >>> # lr = 0.05     if epoch < 30
        >>> # lr = 0.005    if 30 <= epoch < 80
        >>> # lr = 0.0005   if epoch >= 80
        >>> scheduler = MultiStepLR(optimizer, milestones=[30,80], gamma=0.1)
        >>> for epoch in range(100):
        >>>     train(...)
        >>>     validate(...)
        >>>     scheduler.step()