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
- /experiment/slot_attention/_base
- /dataset/cater6_image
- /experiment/slot_attention/_preprocessing_cater
- /experiment/slot_attention/_metrics_clevr