From b6c1cae67b139c6c27112147375b04ca1cfec3f8 Mon Sep 17 00:00:00 2001 From: lukovnikov Date: Mon, 18 Mar 2019 13:32:04 +0100 Subject: [PATCH] branches, optim cosine fix --- pytorch_pretrained_bert/optimization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_pretrained_bert/optimization.py b/pytorch_pretrained_bert/optimization.py index 9a873e221b..eb24c3bd37 100644 --- 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):