6 lines
119 B
YAML
6 lines
119 B
YAML
generator:
|
|
_target_: model.Generator
|
|
in_features: 1024
|
|
ffd_hidden_size: 4096
|
|
num_classes: 5
|
|
attn_layer_num: 4 |