configs.yaml

Reward: !munch.Munch
  mlp_dim: 1024
  model_path: ./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