File size: 228 Bytes
8eadd2d |
1 2 3 4 5 6 7 8 9 10 11 12 |
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 |