Source code for flambe.nlp.transformers

from flambe.nlp.transformers.bert import BERTTextField, BERTEmbeddings, BERTEncoder
from flambe.nlp.transformers.openai import OpenAIGPTTextField, OpenAIGPTEmbeddings, OpenAIGPTEncoder
from flambe.nlp.transformers.optim import AdamW, ConstantLRSchedule
from flambe.nlp.transformers.optim import WarmupConstantSchedule, WarmupLinearSchedule


__all__ = ['BERTTextField', 'BERTEmbeddings', 'BERTEncoder',
           'OpenAIGPTTextField', 'OpenAIGPTEmbeddings', 'OpenAIGPTEncoder',
           'AdamW', 'ConstantLRSchedule', 'WarmupConstantSchedule', 'WarmupLinearSchedule']