From 80dcd02c52b11e944967e5e80a6562c4a8b3ed2e Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 27 Oct 2021 22:43:17 +0200 Subject: Update conv transfromer experiment config --- training/conf/experiment/conv_transformer_paragraphs.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'training/conf') diff --git a/training/conf/experiment/conv_transformer_paragraphs.yaml b/training/conf/experiment/conv_transformer_paragraphs.yaml index 4de9722..8c3af44 100644 --- a/training/conf/experiment/conv_transformer_paragraphs.yaml +++ b/training/conf/experiment/conv_transformer_paragraphs.yaml @@ -50,7 +50,7 @@ lr_schedulers: max_lr: 1.0e-4 total_steps: null epochs: *epochs - steps_per_epoch: 316 + steps_per_epoch: 211 pct_start: 0.03 anneal_strategy: cos cycle_momentum: true @@ -66,7 +66,7 @@ lr_schedulers: datamodule: _target_: text_recognizer.data.iam_extended_paragraphs.IAMExtendedParagraphs - batch_size: 4 + batch_size: 6 num_workers: 12 train_fraction: 0.8 pin_memory: true @@ -75,7 +75,7 @@ datamodule: network: _target_: text_recognizer.networks.conv_transformer.ConvTransformer input_dims: [1, 576, 640] - hidden_dim: &hidden_dim 256 + hidden_dim: &hidden_dim 192 encoder_dim: 1280 dropout_rate: 0.1 num_classes: *num_classes -- cgit v1.2.3-70-g09d2