Source code for flambe.nlp.transformers.optim

"""
Intergation of the pytorch_transformers optimization module
"""

import pytorch_transformers as pt

from flambe import Component



[docs]class ConstantLRSchedule(Component, pt.ConstantLRSchedule):
    pass




[docs]class WarmupConstantSchedule(Component, pt.WarmupConstantSchedule):
    pass




[docs]class WarmupLinearSchedule(Component, pt.WarmupLinearSchedule):
    pass




[docs]class AdamW(Component, pt.AdamW):
    pass