kd5678 commited on
Commit
8eadd2d
1 Parent(s): 87559b6

Upload configs.yaml

Browse files
Files changed (1) hide show
  1. configs.yaml +12 -0
configs.yaml ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ImageReward: !munch.Munch
2
+ mlp_dim: 1024
3
+ model_path: ~/.cache/prefpaintReward/prefpaintReward.pt
4
+ ViT: !munch.Munch
5
+ dropout: 0.1
6
+ feature_dim: 1024
7
+ mlp_dim: 1024
8
+ num_heads: 16
9
+
10
+ fix_base: false
11
+ fix_rate: 0.7
12
+ group: False