Spaces:
Running
Running
File size: 305 Bytes
033bd8b |
1 2 3 4 5 6 7 8 9 10 11 12 |
class LRMult(object):
def __init__(self, lr_mult=1.):
self.lr_mult = lr_mult
def __call__(self, m):
if getattr(m, 'weight', None) is not None:
m.weight.lr_mult = self.lr_mult
if getattr(m, 'bias', None) is not None:
m.bias.lr_mult = self.lr_mult
|