Skip to content

configs/experiment/slot_attention/cater6.yaml

# @package _global_
# Configuration to exactly reproduce unsupervised object recognition of the original slot attention
# paper.
defaults:
  - /experiment/slot_attention/_base  # (1)!
  - /dataset: cater6_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: 7
  1. /experiment/slot_attention/_base
  2. /dataset/cater6_image
  3. /experiment/slot_attention/_preprocessing_cater
  4. /experiment/slot_attention/_metrics_clevr