From 31d58f2108165802d26eb1c1bdb9e5f052b4dd26 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Tue, 6 Apr 2021 22:31:54 +0200 Subject: Fix network args --- training/experiments/image_transformer.yaml | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) (limited to 'training/experiments') diff --git a/training/experiments/image_transformer.yaml b/training/experiments/image_transformer.yaml index 012a19b..9e8f9fc 100644 --- a/training/experiments/image_transformer.yaml +++ b/training/experiments/image_transformer.yaml @@ -1,12 +1,9 @@ network: type: ImageTransformer args: - input_shape: None - output_shape: None encoder: type: None args: None - mapping: sentence_piece num_decoder_layers: 4 hidden_dim: 256 num_heads: 4 @@ -60,7 +57,7 @@ callbacks: trainer: args: stochastic_weight_avg: true - auto_scale_batch_size: power + auto_scale_batch_size: binsearch gradient_clip_val: 0 fast_dev_run: false gpus: 1 -- cgit v1.2.3-70-g09d2