changing docker

eac039d2 · lukovnikov · 471daf1b · eac039d2
Commit eac039d2 authored Mar 12, 2019 by lukovnikov
Hide 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
@@ -24,7 +24,7 @@ import logging
 logger = logging.getLogger(__name__)


-__all__ = ["LRSchedule", "WarmupLinearSchedule", "WarmupConstantSchedule", "WarmupCosineSchedule", "BertAdam"]
+__all__ = ["LRSchedule", "WarmupLinearSchedule", "WarmupConstantSchedule", "WarmupCosineSchedule", "BertAdam", "WarmupCosineWithRestartsSchedule"]


 class LRSchedule(object):