config.yaml

model:
  target: pixelflow.model.PixelFlowModel
  params:
    num_attention_heads: 16
    attention_head_dim: 72
    in_channels: 3
    out_channels: 3
    depth: 28
    num_classes: 1000
    patch_size: 4
    attention_bias: true
scheduler:
  num_train_timesteps: 1000
  num_stages: 4
  pyramid_shift: false
data:
  root: /public/datasets/ILSVRC2012/train
  center_crop: false
  resolution: 256
  expand_ratio: 1.125
  num_workers: 4
  batch_size: 4
seed: 42