activation_checkpointing: false
activation_offloading: false
batch_size: 64
channels_last: true
dataset_dir: .cache/mnist
decoder: !!python/object:perceiver.model.core.classifier.ClassificationDecoderConfig
  cross_attention_residual: true
  cross_attention_widening_factor: 1
  dropout: 0.1
  freeze: false
  init_scale: 0.1
  num_classes: 10
  num_cross_attention_heads: 1
  num_cross_attention_qk_channels: null
  num_cross_attention_v_channels: null
  num_output_queries: 1
  num_output_query_channels: 128
encoder: !!python/object:perceiver.model.vision.image_classifier.ImageEncoderConfig
  cross_attention_widening_factor: 1
  dropout: 0.1
  first_cross_attention_layer_shared: false
  first_self_attention_block_shared: false
  freeze: false
  image_shape: !!python/tuple
  - 28
  - 28
  - 1
  init_scale: 0.1
  num_cross_attention_heads: 1
  num_cross_attention_layers: 2
  num_cross_attention_qk_channels: null
  num_cross_attention_v_channels: null
  num_frequency_bands: 32
  num_self_attention_blocks: 3
  num_self_attention_heads: 8
  num_self_attention_layers_per_block: 3
  num_self_attention_qk_channels: null
  num_self_attention_v_channels: null
  self_attention_widening_factor: 1
normalize: true
num_latent_channels: 128
num_latents: 32
num_workers: 3
params: null
pin_memory: true
random_crop: null
shuffle: true
