mirrors/deep-person-reid
1
0
Fork 0
mirror of https://github.com/KaiyangZhou/deep-person-reid.git synced 2025-06-03 14:53:23 +08:00
Code Issues Projects Releases Wiki Activity

add torchtools

Browse Source
This commit is contained in:
KaiyangZhou 2018-07-04 10:32:32 +01:00
parent 9dd1d48d3c
commit ade721922c
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
utils/torchtools.py Normal file
View File

@ -0,0 +1,10 @@
from __future__ import absolute_import
from __future__ import division
import torch
def adjust_learning_rate(optimizer, base_lr, epoch, stepsize, gamma=0.1):
lr = base_lr * (gamma ** (epoch // stepsize))
for param_group in optimizer.param_groups:
param_group['lr'] = lr