diff --git a/learned_optimization/tasks/fixed/resnet.py b/learned_optimization/tasks/fixed/resnet.py index afbf2b7..0004d85 100644 --- a/learned_optimization/tasks/fixed/resnet.py +++ b/learned_optimization/tasks/fixed/resnet.py @@ -28,7 +28,7 @@ class _ResnetTask(base.Task): - """Tranformer from a dictionary configuration.""" + """Transformer from a dictionary configuration.""" def __init__(self, cfg: Mapping[str, Any], name: str = '__Resnet'): dataset_map = { diff --git a/learned_optimization/tasks/fixed/transformer_lm.py b/learned_optimization/tasks/fixed/transformer_lm.py index 2482442..247e0c5 100644 --- a/learned_optimization/tasks/fixed/transformer_lm.py +++ b/learned_optimization/tasks/fixed/transformer_lm.py @@ -28,7 +28,7 @@ class _TransformerTask(base.Task): - """Tranformer from a dictionary configuration.""" + """Transformer from a dictionary configuration.""" def __init__(self, cfg: Mapping[str, Any], name: str = '__TransformerTask'): self.datasets = language.lm1b_32k_datasets(cfg['batch_size'],