diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-06 22:31:54 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-06 22:31:54 +0200 |
| commit | 31d58f2108165802d26eb1c1bdb9e5f052b4dd26 (patch) | |
| tree | 6f5c2dcb0eef814c71a34df98444be7e8f1d0b43 /training/experiments | |
| parent | 5e11924ca6aaea7898caca94675f41f67706a406 (diff) | |
Fix network args
Diffstat (limited to 'training/experiments')
| -rw-r--r-- | training/experiments/image_transformer.yaml | 5 |
1 files changed, 1 insertions, 4 deletions
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 |