branches, optim cosine fix

b6c1cae6 · lukovnikov · 20e65220 · b6c1cae6
Commit b6c1cae6 authored Mar 18, 2019 by lukovnikov
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

pytorch_pretrained_bert/optimization.py pytorch_pretrained_bert/optimization.py +1 -1

No files found.
--- 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):