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