configs.yaml

ImageReward: !munch.Munch
  mlp_dim: 1024
  model_path: ~/.cache/prefpaintReward/prefpaintReward.pt
ViT: !munch.Munch
  dropout: 0.1
  feature_dim: 1024
  mlp_dim: 1024
  num_heads: 16

fix_base: false
fix_rate: 0.7
group: False