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