branches, optim cosine fix
This commit is contained in:
@@ -52,7 +52,7 @@ class LRSchedule(object):
|
|||||||
|
|
||||||
def get_lr_(self, step):
|
def get_lr_(self, step):
|
||||||
return 1.
|
return 1.
|
||||||
# raise NotImplemented("use subclass")
|
# raise NotImplemented("use subclass") -
|
||||||
|
|
||||||
|
|
||||||
class WarmupCosineSchedule(LRSchedule):
|
class WarmupCosineSchedule(LRSchedule):
|
||||||
|
|||||||
Reference in New Issue
Block a user