configs/experiment/slot_attention/coco.yaml

# @package _global_
defaults:
  - /experiment/slot_attention/_base_large  # (1)!
  - /dataset: coco  # (2)!
  - /experiment/slot_attention/_preprocessing_coco # (3)!
  - /experiment/slot_attention/_metrics_coco # (4)!
  - _self_

trainer:
  devices: 8
  max_steps: 500000
  max_epochs:
  gradient_clip_val: 1.0

dataset:
  num_workers: 4
  batch_size: 8

models:
  conditioning:
    _target_: routed.ocl.conditioning.RandomConditioning
    object_dim: 256
    n_slots: 11
    batch_size_path: input.batch_size