branches, optim cosine fix

2019-03-18 13:32:04 +01:00
parent 20e652209c
commit b6c1cae67b
1 changed files with 1 additions and 1 deletions
--- a/pytorch_pretrained_bert/optimization.py
+++ b/pytorch_pretrained_bert/optimization.py
@@ -52,7 +52,7 @@ class LRSchedule(object):

    def get_lr_(self, step):
        return 1.
-        # raise NotImplemented("use subclass")
+        # raise NotImplemented("use subclass")  -


 class WarmupCosineSchedule(LRSchedule):