Skip to content

configs/experiment/slot_attention/cater10.yaml

# @package _global_
# Configuration to exactly reproduce unsupervised object recognition of the original slot attention
# paper.
defaults:
  - /experiment/slot_attention/_base  # (1)!
  - /dataset: cater_image  # (2)!
  - /experiment/slot_attention/_preprocessing_cater # (3)!
  - /experiment/slot_attention/_metrics_clevr # (4)!
  - _self_

trainer:
  devices: 8
  max_steps: 500000
  max_epochs:
dataset:
  num_workers: 4
  batch_size: 8
models:
  conditioning:
    n_slots: 11
  1. /experiment/slot_attention/_base
  2. /dataset/cater_image
  3. /experiment/slot_attention/_preprocessing_cater
  4. /experiment/slot_attention/_metrics_clevr