flambe.nlp.transformers.optim
¶
Intergation of the pytorch_transformers optimization module
Module Contents¶
-
class
flambe.nlp.transformers.optim.
ConstantLRSchedule
[source]¶ Bases:
flambe.Component
,pytorch_transformers.ConstantLRSchedule
-
class
flambe.nlp.transformers.optim.
WarmupConstantSchedule
[source]¶ Bases:
flambe.Component
,pytorch_transformers.WarmupConstantSchedule
-
class
flambe.nlp.transformers.optim.
WarmupLinearSchedule
[source]¶ Bases:
flambe.Component
,pytorch_transformers.WarmupLinearSchedule
-
class
flambe.nlp.transformers.optim.
AdamW
[source]¶ Bases:
flambe.Component
,pytorch_transformers.AdamW