Upload eval-model/eval.yaml with huggingface_hub
Browse files- eval-model/eval.yaml +6 -0
eval-model/eval.yaml
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
generator:
|
| 2 |
+
_target_: pretrained.eval.Generator
|
| 3 |
+
in_features: 1024
|
| 4 |
+
ffd_hidden_size: 4096
|
| 5 |
+
num_classes: 5
|
| 6 |
+
attn_layer_num: 4
|