summaryrefslogtreecommitdiff
path: root/training
diff options
context:
space:
mode:
Diffstat (limited to 'training')
-rw-r--r--training/experiments/image_transformer.yaml12
-rw-r--r--training/run_experiment.py4
2 files changed, 9 insertions, 7 deletions
diff --git a/training/experiments/image_transformer.yaml b/training/experiments/image_transformer.yaml
index 7f0bbb7..012a19b 100644
--- a/training/experiments/image_transformer.yaml
+++ b/training/experiments/image_transformer.yaml
@@ -1,6 +1,6 @@
network:
type: ImageTransformer
- args:
+ args:
input_shape: None
output_shape: None
encoder:
@@ -17,20 +17,20 @@ network:
model:
type: LitTransformerModel
args:
- optimizer:
+ optimizer:
type: MADGRAD
args:
lr: 1.0e-2
momentum: 0.9
weight_decay: 0
eps: 1.0e-6
- lr_scheduler:
+ lr_scheduler:
type: CosineAnnealingLR
- args:
+ args:
T_max: 512
criterion:
type: CrossEntropyLoss
- args:
+ args:
weight: None
ignore_index: -100
reduction: mean
@@ -40,7 +40,7 @@ model:
data:
type: IAMExtendedParagraphs
- args:
+ args:
batch_size: 16
num_workers: 12
train_fraction: 0.8
diff --git a/training/run_experiment.py b/training/run_experiment.py
index 8a29555..0a67bfa 100644
--- a/training/run_experiment.py
+++ b/training/run_experiment.py
@@ -50,7 +50,9 @@ def _import_class(module_and_class_name: str) -> type:
return getattr(module, class_name)
-def _configure_pl_callbacks(args: List[Union[OmegaConf, NamedTuple]]) -> List[Type[pl.callbacks.Callback]]:
+def _configure_pl_callbacks(
+ args: List[Union[OmegaConf, NamedTuple]]
+) -> List[Type[pl.callbacks.Callback]]:
"""Configures PyTorch Lightning callbacks."""
pl_callbacks = [
getattr(pl.callbacks, callback.type)(**callback.args) for callback in args